Party at Hali-Bula UVA - 1220 树的最大独立集

紫书dp

 

#include<bits/stdc++.h>
using namespace std;
int n;
int cnt=0;
int d[205][3],f[205][3];//d(u,0) f(u,0)标志以u为根的子树中不选u点所能得到的最大人数以及方案的唯一性  f(u,0)=k  k==1唯一  0不唯一而d(u,1) f(u,1)则是取u点的最大人数 
vector<int>sons[205];
map<string ,int>dict;
int ID(const string& s) {
  if(!dict.count(s)) dict[s] = cnt++;
  return dict[s];
}
int dp(int u,int k){
	f[u][k]=1;//初始化为方案唯一	
	d[u][k]=k;	//如果选k==1则  至少有选的改点初始化为d(u,0)最大人数为一
	for(int i=0;i<sons[u].size();i++){
		int v=sons[u][i];
		if(k==1){//选择该点时，加上不选子节点的最大值
			d[u][k]+=dp(v,0);
			if(!f[v][0])f[u][k]=0;//如果子节点选大取法不唯一 则 父节点的最大取法也不唯一
		}
		else {//不选该点时  子节点可选可不选  加上两者的最大值
			d[u][k]+=max(dp(v,0),dp(v,1));
			if(d[v][0]==d[v][1]){//如果选子节点和不选子节点的最大数相同则以u为根节点的最大数肯定也相同
				f[u][k]=0;
			}
			else if(d[v][0]>d[v][1]&&!f[v][0])f[u][k]=0;//如果不选v大于选v的最大人数并且不选v的方案不唯一 则让v的父节点到达最大的方案也不唯一 下同
			else if(d[v][1]>d[v][0]&&!f[v][1])f[u][k]=0;

		}
	}
	return d[u][k];
}

int main(){
		string s1,s2,s;
		while(cin>>n&&n){
			cin>>s;
			cnt=0;
			dict.clear();
			for(int i=0;i<n;i++)sons[i].clear();
			ID(s);
			for(int i=1;i<n;i++){
                    cin>>s1>>s2;
					sons[ID(s2)].push_back(ID(s1));
			}
			printf("%d ",max(dp(0,0),dp(0,1)));
			bool flag=0;//以0为根节点选0大于不选0并且方案唯一 下同
			if(d[0][0] > d[0][1] && f[0][0]) flag= true;
        if(d[0][1] > d[0][0] && f[0][1]) flag = true;
        if(flag) printf("Yes\n"); 
		else printf("No\n");
		}
	return 0;
}