Codeforces 369C Valera And Elections 树形DP

点击打开链接

题意:n个结点的树,n<=1e5 某些边(u,v)需要修理,若选中v,则v->1路径上的边都被修理,求选出最少点使得要修理的边都被修理

容易得到,若u->1,v->1都有要修理的边,&&u是v的祖先,则选中v即可,dp算出子树中是否有边需要被修理,一个点i需要被选中,只有当边(u,i)=2&&子树没有边需要被修理

#include <bits/stdc++.h>using namespace std;typedef long long ll;typedef pair<int,int> ii;const int N=2e5+20; int n,a[N];int dp[N];//dp[i] 结点i的子树中是否有边需要被修理 vector<ii> e[N];vector<int> ans;void dfs(int u,int fa){for(int i=0;i<e[u].size();i++){int v=e[u][i].first;int k=e[u][i].second;if(v!=fa){if(k==2){dp[u]=1;//u子树中有要修理的边 a[v]=1;//v为u的son,&&(u,v)边需要修理 }dfs(v,u);if(dp[v])dp[u]=1;}}}int main(){int n;while(cin>>n){memset(a,0,sizeof(a));memset(dp,0,sizeof(dp));int u,v,k;for(int i=1;i<=n;i++)e[i].clear();for(int i=1;i<=n-1;i++){//题目给出的边可能不是fa->son 如果是son->fa&&只标记a[v],则会出错 scanf("%d%d%d",&u,&v,&k);e[u].push_back(make_pair(v,k));e[v].push_back(make_pair(u,k));}dfs(1,0);ans.clear();for(int i=1;i<=n;i++){if(a[i]==1&&dp[i]==0)ans.push_back(i);}cout<<ans.size()<<endl;for(int i=0;i<ans.size();i++){cout<<ans[i];if(i==ans.size()-1)cout<<endl;elsecout<<' ';}}return 0;}