DTOJ #3316. baka
【题目描述】
给定一颗有方向的树,改变一条边方向的代价为 $1$,求使得从两个点出发能到达树上所有节点的最小代价。
【输入格式】
第一行一个正整数 $n$。
接下来 $n-1$ 行,每行两个正整数 $u,v$,表示 $u$ 向 $v$ 连一条有向边。
【输出格式】
一行一个数表示答案。
【样例】
样例输入
4
2 1
3 1
4 1
样例输出
1
【数据范围与提示】
对于 $100\%$ 的数据,$1\leq n\leq 10^6]$。
【题解】
本题有两种思路。
第一种,考虑到有两个起点,设 $f[i][1/2][0/1]$ 表示以 $i$ 为根的子树内已经有 $1/2$ 个出发点,$i$ 是否为出发点,直接转移即可。
第二种,考虑到两个起点必然将整棵树分成两部分,考虑断边,换根 $dp$ 即可。
【代码】
#include<bits/stdc++.h> const int maxn=1000000+10; const int inf=1000000000; int n,f[maxn][2][2],g[2][2]; std::vector<int> E[maxn],G[maxn]; inline int read ( void ) { int x=0;char ch;bool f=true; while ( !isdigit(ch=getchar()) ) if ( ch=='-' ) f=false; for ( x=ch^48;isdigit(ch=getchar()); ) x=(x<<1)+(x<<3)+(ch^48); return f ? x : -x ; } using std::min; inline void dfs ( int u,int fr ) { f[u][0][0]=f[u][1][0]=f[u][1][1]=inf; for ( int v:E[u] ) if ( v!=fr ) { dfs(v,u); for ( int i=0;i<2;i++ ) for ( int j=0;j<2;j++ ) g[i][j]=f[u][i][j]; f[u][0][0]=min(g[0][0]+f[v][0][1],g[0][1]+min(f[v][0][0],f[v][0][1])+1); f[u][0][1]=g[0][1]+f[v][0][1]; f[u][1][0]=min(min(g[0][0]+min(min(f[v][0][0],f[v][1][1]),f[v][0][1]+1),g[0][1]+min(f[v][1][0],f[v][1][1])+1),min(g[1][0]+f[v][0][1],g[1][1]+min(f[v][0][0],f[v][0][1])+1)); f[u][1][1]=min(g[1][1]+f[v][0][1],g[0][1]+min(f[v][1][1],f[v][0][0])); } for ( int v:G[u] ) if ( v!=fr ) { dfs(v,u); for ( int i=0;i<2;i++ ) for ( int j=0;j<2;j++ ) g[i][j]=f[u][i][j]; f[u][0][0]=min(g[0][0]+f[v][0][1]+1,g[0][1]+min(f[v][0][0],f[v][0][1])); f[u][0][1]=g[0][1]+f[v][0][1]+1; f[u][1][0]=min(min(g[0][0]+min(min(f[v][0][0],f[v][1][1]+1),f[v][0][1]+0),g[0][1]+min(f[v][1][0],f[v][1][1])+0),min(g[1][0]+f[v][0][1]+1,g[1][1]+min(f[v][0][0],f[v][0][1]))); f[u][1][1]=min(g[1][1]+f[v][0][1],g[0][1]+min(f[v][1][1],f[v][0][0]))+1; } } signed main() { n=read(); for ( int i=1,u,v;i<n;i++ ) u=read(),v=read(),E[u].push_back(v),G[v].push_back(u); dfs(1,0); return !printf("%d\n",min(min(f[1][0][0],f[1][0][1]),min(f[1][1][0],f[1][1][1]))); }