Hackrank Kingdom Division 树形DP

题目链接：传送门

题意：

　　给你一棵树，n个点

　　每个点可以染成红色和蓝色

　　但是红色的点与其相邻的点中必须有红色节点，蓝色也是

　　问你有多少种染色的方案

题解：

　　树形dp

　　先转化为有根树，取1为根

　　设定dp[now][red][red] 表示的是当前now节点然red色，其父亲节点染red色的可行方案数

　　转移很容易想到

#include<bits/stdc++.h>
using namespace std;
#pragma comment(linker, "/STACK:102400000,102400000")
#define ls i<<1
#define rs ls | 1
#define mid ((ll+rr)>>1)
#define pii pair<int,int>
#define MP make_pair
typedef long long LL;
const long long INF = 1e18+1LL;
const double Pi = acos(-1.0);
const int N = 5e5+, M = 1e2+,inf = 2e9;
const LL mod = 1e9+;
 
LL dp[N][][];
vector<int > G[N];
int n;
void dfs(int u,int f) {
    for(int i = ; i < G[u].size(); ++i) {
        int to = G[u][i];
        if(to == f) continue;
        dfs(to,u);
    }
    dp[u][][] = ;
    dp[u][][] = ;
    int ok = ;
    LL ans1=  ,ans2 = ;
    for(int i = ; i < G[u].size(); ++i) {
        int to = G[u][i];
        if(to == f) continue;
        ok = ;
        dp[u][][] *= (dp[to][][]+dp[to][][])%mod;
        dp[u][][] %= mod;
 
        dp[u][][] *= (dp[to][][]+dp[to][][])%mod;
        dp[u][][] %= mod;
        ans1 *= (dp[to][][]) % mod;
        ans1 %= mod;
        ans2 *= (dp[to][][]) % mod;
        ans2 %= mod;
    }
    if(ok) {
        dp[u][][] = (dp[u][][] - ans1 + mod) % mod;
        dp[u][][] = (dp[u][][] - ans2 + mod) % mod;
    }
}
 
int main() {
    scanf("%d",&n);
    for(int i = ; i < n; ++i) {
        int x,y;
        scanf("%d%d",&x,&y);
        G[x].push_back(y);
        G[y].push_back(x);
    }
    dfs(,);
    printf("%lld\n",(dp[][][]+dp[][][]) % mod);
    return ;
}