第一道树形DP
详细见分析。。
View Code
1 #include<stdio.h> 2 #include<string> 3 #include<stdlib.h> 4 #include<map> 5 #include<algorithm> 6 #include<iostream> 7 using namespace std; 8 const int maxn = 205; 9 const int maxm = 40005; 10 struct node{ 11 int u,v,next; 12 }edge[ maxm ]; 13 int cnt ,head[ maxn ]; 14 map<string,int>mp; 15 16 void init(){ 17 cnt = 0; 18 memset( head,-1,sizeof( head ) ); 19 mp.clear(); 20 } 21 22 void addedge( int a,int b ){ 23 edge[ cnt ].u = a; 24 edge[ cnt ].v = b; 25 edge[ cnt ].next = head[ a ]; 26 head[ a ] = cnt++; 27 } 28 29 int dp[ maxn ][ 2 ];//dp[i][0]:第i个节点不去聚会的最多人数，dp[i][1]:第i个节点去聚会的最多人数 30 //dp[i][0] = sigma of sons ( max(dp[j][0],dp[j][1]) ); 31 //dp[i][1] = sigma of sons ( dp[j][0] ); 32 int dup[ maxn ][ 2 ]; 33 //dup[i][j]=0:表示对于dp[i][j]这么多人参加聚会时，方案不是独一无二的。 34 //dup[i][j]=1:表示对于dp[i][j]这么多人参加聚会时，方案是独一无二的。 35 //如果儿子去和不去是一样的，那么父亲不去的话,肯定存在多种方案。如果父亲去的话，维持原状。 36 //如果儿子不去的话存在多种方案，那么父亲去的话也会存在多种方案（因为父亲的值有儿子传过来，儿子不去有多种方案说明再加上一个父亲还是有多种方案） 37 38 39 void dfs( int now ){ 40 dp[ now ][ 0 ] = 0; 41 dp[ now ][ 1 ] = 1; 42 43 dup[ now ][0]=1; 44 dup[ now ][1]=1; 45 46 for( int i=head[ now ];i!=-1;i=edge[ i ].next ){ 47 int v = edge[ i ].v; 48 dfs( v ); 49 dp[ now ][ 0 ] += max( dp[ v ][ 0 ],dp[ v ][ 1 ] ); 50 dp[ now ][ 1 ] += dp[ v ][ 0 ]; 51 52 if( dp[ v ][0]==dp[ v ][1] ) dup[ now ][ 0 ] =0; 53 if( dup[ v ][0]==0 ) dup[ now ][1]=0; 54 } 55 } 56 57 int main(){ 58 int n; 59 while( scanf("%d",&n)==1&&n ){ 60 string s1,s2; 61 cin>>s1; 62 init(); 63 int pcnt = 1; 64 mp[ s1 ] = pcnt++; 65 int num1,num2; 66 for( int i=0;i<n-1;i++ ){ 67 cin>>s1>>s2; 68 69 if( mp[s1]==0 ){ 70 num1 = pcnt; 71 mp[s1]=pcnt++; 72 } 73 else num1 = mp[s1]; 74 75 if( mp[s2]==0 ){ 76 num2 = pcnt; 77 mp[s2]=pcnt++; 78 } 79 else num2 = mp[s2]; 80 81 addedge( num2,num1 ); 82 } 83 memset( dp,0,sizeof( dp ) ); 84 dfs( 1 );//树形DP 85 if( n==1 ){ 86 printf("1 Yes\n"); 87 continue; 88 } 89 /* 90 int ans = max( dp[1][0],dp[1][1] ); 91 int flag = 1; 92 if( dp[1][0]==dp[1][0] ) flag = -1; 93 for( int i=2;i<maxn&&flag==1;i++ ){ 94 if( dp[i][0]==ans || dp[i][1]==ans ){ 95 flag = -1; 96 break; 97 } 98 } 99 if( flag==-1 ) printf("%d No\n",ans); 100 else printf("%d Yes\n",ans); 101 //不懂为什么这种方法会错。。。 102 */ 103 104 if(dp[1][0]>dp[1][1] && dup[1][0]) printf("%d Yes\n",dp[1][0]); //yes表示是独一无二的 105 else if(dp[1][1]>dp[1][0] && dup[1][1]) printf("%d Yes\n",dp[1][1]); 106 else printf("%d No\n",max(dp[1][0],dp[1][1])); 107 108 } 109 return 0; 110 }