luogu 4084

 给树的点染色，颜色有3种，相邻的点颜色不同

某些点已经染色

方案数？

#include <bits/stdc++.h>
 using namespace std ;
 const int N=1e5+2,M=5*N;
 
 #define int long long
 const int mod=1e9+7;
 
  int f[N][4],n;
  int nxt[M],go[M],hd[N],all;
 
 void add(int x,int y){
 	go[++all]=y,nxt[all]=hd[x];
	hd[x]=all;
 }
  void dfs(int x,int fa){
  	int i;
  	
  	for(i=1;i<=3;i++){	  		
	   if(f[x][i]){
			for(int j=1;j<i;j++) f[x][j]=0;break;
	    } 
	 f[x][i]=1;
	}
	
  	for(i=hd[x];i;i=nxt[i]){
  		int y=go[i]; if(y==fa) continue; dfs(y,x);
  		
  		f[x][1]=f[x][1]*(f[y][2]+f[y][3]);
  		f[x][2]=f[x][2]*(f[y][1]+f[y][3]);
  		f[x][3]=f[x][3]*(f[y][1]+f[y][2]);
  		f[x][1]%=mod;f[x][2]%=mod;f[x][3]%=mod;
	}
  }
  signed main(){
  	int i,x,y,K;
  	cin>>n>>K;
  	
  	for(i=1;i<n;i++) cin>>x>>y,add(x,y),add(y,x);
	for(i=1;i<=K;i++) cin>>x>>y,f[x][y]=1;
	
	dfs(1,0);
	cout<<(f[1][1]+f[1][2]+f[1][3])%mod<<endl;
  }