Codeforces Round #747 (Div. 2)E2. Rubik‘s Cube Coloring (hard version)（记忆化+dfs+dp）

Codeforces Round #747 (Div. 2)E2. Rubik’s Cube Coloring (hard version)（记忆化+dfs+dp）

题目链接：E2. Rubik’s Cube Coloring (hard version)

题意：

给定一颗高度为$k$的满二叉树，其中$n$个节点已经给定了指定颜色，让你给其他每个节点染色，问有多少种染色方法使得满足如下条件：
白色节点不能与白色和黄色节点相邻
黄色节点不能与白色和黄色节点相邻
绿色节点不能与绿色和蓝色节点相邻
蓝色节点不能与绿色和蓝色节点相邻
红色节点不能与红色和橙色节点相邻
橙色节点不能与红色和橙色节点相邻

题解：

我们可以把一整棵树中的子树分为两种，一种是每个根节点的子树中没有被染色点，另一种是每个根节点的子树中包含被染色点。
首先考虑第一种：
对于一个子树中没有被染色点的树，显然根节点染什么颜色对应的方案数是相同的,因此我们令$all[i]$表示$i$层满二叉树无已染色儿子的方案数(根节点已确定为某种颜色),显然$all[1]=1$,对于每个点，其左右儿子可以各选$4$种颜色，可推出状态转移方程方程：
$all[i]=4\ast all[i-1]\ast 4\ast all[i-1]=16\ast all[i-1{]}^2$
因此我们可以预处理出$all$数组

对于第二种：
我们可以利用$dfs$枚举其左右儿子的颜色进行转移，直到其子树中无染色节点，转化成第一种。

在$dfs$中我们需要枚举当前节点和对应颜色，这样当点实际颜色与枚举颜色不同时可以直接返回，子树中无染色点时也可以通过$all$数组直接求解，避免了枚举大量无用情况，并且因为子树中有染色点的点最多只有$2000\ast 60$个不到，我们可以通过记忆化来优化。在实现时可以将颜色映射成编号，把不能相邻的颜色用二维数组表示
代码中有很多技巧，具体可以看代码和注释，注释写的还是很详细的

ll all[110];//i层满二叉树无已染色儿子的方案数(根节点已确定为某种颜色)
map<string,int>mp;//颜色映射编号
map<ll,int>color;//当前点是否被染色
map<ll,int>son;//当前点儿子是否被染色
int vis[10][10];
void init(int dep){
	mp["white"]=0;//将颜色编号
	mp["yellow"]=1;
	mp["green"]=2;
	mp["blue"]=3;
	mp["red"]=4;
	mp["orange"]=5;
	vis[0][0]=vis[0][1]=1;//不合法的相邻情况
    vis[1][0]=vis[1][1]=1;
    vis[2][3]=vis[2][2]=1;
    vis[3][2]=vis[3][3]=1;
    vis[4][5]=vis[4][4]=1;
    vis[5][4]=vis[5][5]=1;
	all[1]=1;
	for(int i=2;i<=dep;i++){
		all[i]=((16ll*all[i-1])%mod)*all[i-1]%mod;//两个儿子各可以选4种(不和当前点冲突的)颜色
	}
}
ll dp[2100*61][7];//只需要对子树中有染色的点的点记忆化，因为其他点都可以通过all直接返回
int k,n;
ll dfs(ll root,int col,int dep){//当前根标号，对应颜色，深度
	if(color.count(root)&&color[root]!=col){//当前节点已染色，且和枚举的颜色不同，显然不合法
		return 0;
	}
	if(!son.count(root)){//当前点儿子中没有被染色的点
		return all[k-dep+1];
	}
	//除了上面的情况，就只剩下当前点有被染色儿子的情况了
	if(dp[son[root]][col]!=-1){//记忆化
		return dp[son[root]][col];
	}
	ll res=0;
	for(int i=0;i<6;i++){//枚举左儿子点染什么颜色
		if(!vis[col][i]){//当前颜色合法
			for(int j=0;j<6;j++){//枚举右儿子点染什么颜色
				if(!vis[col][j]){//当前颜色合法
					(res+=dfs(2*root,i,dep+1)*dfs(2*root+1,j,dep+1)%mod)%=mod;
				}
			}
		}
	}
	dp[son[root]][col]=res;
	return res;
}
int main(){
/*cout<<setiosflags(ios::fixed)<<setprecision(8)<<ans<<endl;//输出ans（float）格式控制为8位小数（不含整数部分）*/
/*cout<<setprecision(8)<<ans<<endl;//输出ans（float）格式控制为8位小数（含整数部分）*/
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);//同步流
	memset(dp,-1,sizeof(dp));
	cin>>k>>n;
	init(k);//预处理
	int f=0;
	int cnt=0;
	for(int i=1;i<=n;i++){
		ll idx;
		string s;
		cin>>idx>>s;
		color[idx]=mp[s];//标记当前点染的什么颜色
		if(idx!=1&&color.count(idx/2)&&vis[color[idx/2]][color[idx]]){//已染色的点出现非法组合
			f=1;
		}
		idx/=2;
		while(idx>=1){//把父节点都标记，表示其子树中有已被染色的点
			if(son.count(idx)){//优化(当前的如果已被标记，其父亲必定已被标记)
				break;
			}
			son[idx]=++cnt;//将每个子树中有已被染色的点的点编号，方便后面记忆化搜索
			idx/=2;
		}
	}
	if(f){//染色点出现非法情况，直接结束
		cout<<0<<endl;
		return 0;
	}
	ll ans=0;
	for(int i=0;i<6;i++){//枚举根节点颜色
		ans=(ans+dfs(1,i,1))%mod;
	}
	cout<<ans<<endl;
	return 0;
}