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;
			else
			cout<<' ';
		}
	}
	return 0;
}