poj3342树形dp+深度搜索并确定唯一性

//树形dp入门题目
//使得每2个人之间不能有直接的上下级的关系，求最多能选多少个人出来，并且求出获得最大人数的选人方案是否唯一
//唯一性通过状态数组递归确定
//dp[i][0] = ∑max(dp[j][0], dp[j][1]) (j是i的儿子)
//	dp[i][1] = 1 + ∑dp[j][0] (j是i的儿子) 

//268K	32MS
#include <iostream>
#include <map>
#include <string>
#include <vector>
using namespace std;

#define  MAX_N 206
static int n;
static int dp[MAX_N][2];//表示第i个节点选取或者不选取的时候最大的总数
static int dup[MAX_N][2];//表示当前结点的状态是否唯一

//建立树结构
struct  edge
{
	edge(int f,int to):u(f),next(to)
	{

	}
	int u;//起点
	int next;//终点
};

vector<edge> Tree[MAX_N];//表示整棵树

static void initTree()
{
	for (int i=0;i<MAX_N;++i){
	  Tree[i].clear();
	  dup[i][0] = dup[i][1] = 1;
	}
}

//添加x索引的父节点是y
static void addNode(int x,int y)
{
	vector<edge>&vec = Tree[y];
	vec.push_back(edge(y,x));
}

static void DFS(int u)
{
	if (Tree[u].size()==0)//叶子结点
	{
		dp[u][1] = 1;
		dp[u][0] = 0;
		dup[u][1] = 1;
		dup[u][0] = 1;
		return;
	}

	for (int i=0;i<Tree[u].size();++i)//遍历所有边
	{
		int t = Tree[u][i].next;
		DFS(t);
		dp[u][0] +=max(dp[t][0],dp[t][1]);
		dp[u][1] +=dp[t][0];
		if (dup[t][0]==0)
		 dup[u][1] = 0;//若不取孩子不唯一那么当前取根显然不唯一
		//对于不取根的情况
		if (dp[t][0]>dp[t][1]&&dup[t][0]==0)
		 dup[u][0] = 0;
		else if (dp[t][0]<dp[t][1]&&dup[t][1]==0)
		 dup[u][0] = 0;
		else if (dp[t][0]==dp[t][1])
		 dup[u][0] = 0;
	}
	dp[u][1]+=1;//只加一次
}

int main()
{
	string str,pstr;
	while(cin>>n&&n)
	{
		map<string,int> vstr;
		map<int,string>par;
		memset(dp,0,sizeof(dp));
		initTree();
		cin>>str;
		vstr[str] = 0;

		for (int i=1;i<n;++i){
		 cin>>str>>pstr;
		 vstr[str] = i;
		 par[i] = pstr;
		}
		//加入结点
		for (int i=1;i<n;++i)
		{
			string p = par[i];
			addNode(i,vstr[p]);
		}


		DFS(0);
		char ret[12];
		if (dp[0][0]>dp[0][1])
		{
			if (dup[0][0]==1)
			 strcpy(ret,"Yes");
			else
			 strcpy(ret,"No");
			printf("%d %s\n",dp[0][0],ret);
			
		}
		else if (dp[0][0]<dp[0][1])
		{
			if (dup[0][1]==1)
			  strcpy(ret,"Yes");
			else
			 strcpy(ret,"No");
			printf("%d %s\n",dp[0][1],ret);
		}
		else
			printf("%d No\n",dp[0][1]);

	}

	return 0;
}