福建省2014ACM省赛H Big castle (树形dp)

题意：

给出一棵树，然后每个节点有一个状态0或者1，当操作某个节点时，相邻的节点会反转。求能否通过操作使整个树的节点全部变成1.

题解：

开始的想法是dp[u][2] 1表示这个节点是亮的，0表示这个节点是暗的。发现很多状态无法确定，然后多加一维的状态，dp[u][2][2],第二维表示这个节点是否操作过。这个转移的思想就是保证儿子节点全部暗或者全部亮，然后判断这个节点原来亮暗状态然后进行状态转移。要记录的东西主要是：儿子节点暗或者亮的个数，节点亮或者暗并且只有操作过的节点的个数，节点亮或者和暗操作过和没操作过都要的节点个数。我将不清楚，看代码。

注意一个坑点，有可能输入会超市，用自读入。

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#include<queue>
#include<map>
#include<set>
using namespace std;
#define B(x) (1LL<<(x))
void cmax(int& a,int b){ if(b>a)a=b; }
void cmin(int& a,int b){ if(b<a)a=b; }
typedef long long ll;
const int oo=0x3f3f3f3f;
const int MOD=1000000007;
const int maxn=50005;
int dp[maxn][2][2];
int val[maxn];
struct EDGE{
    int v,next;
}E[maxn<<1];
int head[maxn],tol;

void Init(){
    memset(head,-1,sizeof head);
    tol=0;
}

void add_edge(int u,int v){
    E[tol].v=v;
    E[tol].next=head[u];
    head[u]=tol++;
}

void tree_dp(int u,int pre){

    dp[u][0][0]=dp[u][0][1]=0;
    dp[u][1][0]=dp[u][1][1]=0;
    int son[2]={0},sum=0;
    int op[2][2]={0};
    for(int i=head[u];i!=-1;i=E[i].next){
        int v=E[i].v;
        if(v==pre)continue;
        tree_dp(v,u);
        sum++;
        if(dp[v][1][0]||dp[v][1][1]){
            son[1]++;
            if(!dp[v][1][0]&&dp[v][1][1])op[1][1]++;
            if(dp[v][1][0]&&dp[v][1][1])op[1][0]++;
        }
        if(dp[v][0][0]||dp[v][0][1]){
            son[0]++;
            if(!dp[v][0][0]&&dp[v][0][1])op[0][1]++;
            if(dp[v][0][1]&&dp[v][0][1])op[0][0]++;
        }
    }
    if(son[1]==sum){
        if(op[1][0]>0){
            dp[u][0][0]=1;
            dp[u][1][0]=1;
        }else{
            if(op[1][1]%2==0&&val[u]==0||op[1][1]%2==1&&val[u]==1)
                dp[u][0][0]=1;
            if(op[1][1]%2==0&&val[u]==1||op[1][1]%2==1&&val[u]==0)
                dp[u][1][0]=1;
        }
    }
    if(son[0]==sum){
        if(op[0][0]>0){
            dp[u][0][1]=1;
            dp[u][1][1]=1;
        }else{
            if(op[0][1]%2==0&&val[u]==1||op[0][1]%2==1&&val[u]==0)
                dp[u][0][1]=1;
            if(op[0][1]%2==0&&val[u]==0||op[0][1]%2==1&&val[u]==1)
                dp[u][1][1]=1;
        }
    }
}

void read(int &num)
{
    char in;
    bool neg=false;
    while(((in=getchar()) > '9' || in<'0') && in!='-') ;
    if(in=='-')
    {
        neg=true;
        while((in=getchar()) >'9' || in<'0');
    }
    num=in-'0';
    while(in=getchar(),in>='0'&&in<='9')
        num*=10,num+=in-'0';
    if(neg)
        num=0-num;
}

int main(){
/*
#define ON 1
#ifdef ON
    freopen("E:\\read.txt","r",stdin);
#endif // ON
//*/

    int n,u,v;
    while(scanf("%d",&n)!=EOF){
        Init();
        for(int i=1;i<=n;i++)read(val[i]);
        for(int i=1;i<=n-1;i++){
            read(u);read(v);
            add_edge(u,v);
            add_edge(v,u);
        }
        tree_dp(1,-1);
        if(dp[1][1][0]||dp[1][1][1]) printf("Great Cdfpysw!\n");
        else printf("Poor Nanaya!\n");
    }
	return 0;
}
/**

*/