题目:http://codeforces.com/contest/219/problem/D
题意:给你一棵树,选择一个结点使得可以改变最少的边的方向达到其他所有点
输出改变边的个数,输出结点(多个的话升序输出)
思路:建边,方向对应0,反边对应为1,先dfs统计结点的子树贡献,
dp[i]表示从i出发在i的子树中的总贡献,
然后dfs统计结点的父亲贡献,
f[i]表示从i出发在的向父节点方向的总贡献,从i的父节点走的话可能
继续向父节点走,也可能向i的兄弟树走,所以f[v] = f[u] + (1-w) + (dp[u] - (dp[v]+w))
最后找最小值
这个代码用的f[v] = f[u] + (1-w) - w没怎么理解,没想清楚怎么加的
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 2*1e5+5;
struct edge
{
int v,w;
};
vector<edge> G[N];
vector<int> V;
int dp[N],f[N];
void dfs1(int u,int fa)
{
for(auto e : G[u])
{
int v = e.v,w = e.w;
if(v == fa)
continue;
dfs1(v,u);
dp[u] += dp[v] + w;
}
}
void dfs2(int u,int fa)
{
for(auto e : G[u])
{
int v = e.v,w = e.w;
if(v == fa)
continue;
f[v] = f[u] + (1-w) + (dp[u] - (dp[v]+w));
dfs2(v,u);
}
}
int main()
{
int n;
scanf("%d",&n);
int u,v;
for(int i = 1;i < n;i++)
{
scanf("%d%d",&u,&v);
G[u].push_back({v,0});
G[v].push_back({u,1});
}
dfs1(1,-1);
dfs2(1,-1);
for(int i = 1;i <= n;i++)
f[i] += dp[i];
int minn = *min_element(f+1,f+n+1);
printf("%d\n",minn);
for(int i = 1;i <= n;i++)
if(f[i] == minn)
printf("%d ",i);
printf("\n");
return 0;
}