bzoj5141 [Usaco2017 Dec]Barn Painting（树形dp）

给定一棵n个节点的树，给每个节点染色，一共3种颜色，其中m个节点已经染好色了。一种合法的染色方案定义为任意两个相邻点异色。

如果我们给根确定了颜色，那么他的子树的其他节点就可以独立的去继续染色。

因此我们可以进行dp。

dp[x][c]表示以x为根的子树，x的颜色为c的染色方案数。

我们有转移$dp[x][c]=\prod\limits_{y}\sum\limits_{c' \neq c}dp[y][c']$，其中y为x的儿子。

如果一个节点已经有颜色c了，那么其他的颜色的方案数为0.

复杂度$O(n)$

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
#define mod 1000000007
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline ll read(){
    ll x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,h[N],num=0,col[N],dp[N][3];
struct edge{
    int to,next;
}data[N<<1];
void dfs(int x,int Fa){
    ll tmp[3]={1,1,1};
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==Fa) continue;dfs(y,x);
        tmp[0]=tmp[0]*(dp[y][1]+dp[y][2])%mod;
        tmp[1]=tmp[1]*(dp[y][0]+dp[y][2])%mod;
        tmp[2]=tmp[2]*(dp[y][0]+dp[y][1])%mod;
    }if(col[x]) dp[x][col[x]-1]=tmp[col[x]-1];
    else for(int i=0;i<3;++i) dp[x][i]=tmp[i];
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();
    for(int i=1;i<n;++i){
        int x=read(),y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }while(m--){int x=read();col[x]=read();}dfs(1,0);
    int ans=0;
    for(int i=0;i<3;++i) ans=(ans+dp[1][i])%mod;
    printf("%d\n",ans);
    return 0;
}