做出来了但是没交
题意
白色节点不能与白色和黄色节点相邻;
黄色节点不能与白色和黄色节点相邻;
绿色节点不能与绿色和蓝色节点相邻;
蓝色节点不能与绿色和蓝色节点相邻;
红色节点不能与红色和橙色节点相邻;
橙色节点不能与红色和橙色节点相邻;
据此给一个高为n((n<60))的完全二叉树染色。求方案数:
不难推出公式: 设(fi)表示以高度为i的树,确定根节点颜色,有多少方案, (f_i=f_i^2*16)(根节点确定, 每个子节点有4种可行颜色)
E2在E1的基础上, 有M个点的的颜色已经给出, 求给其他点染色的方案数。
题解
我们先把E1的(f_{i, j})数组预处理出来(高度为i的树没有预给定颜色, 且根节点颜色为j的方案数)
考虑直接dp:如果这个节点的子树中没有预定义节点, 直接返回f数组, 否则直接暴力dp
此时需要处理的节点个数为子树中包含预定义节点的节点个数, 可以证明这样的点数量为(O(nm))(每个预定义点都在叶子节点最大)
现在的问题是如何判断一个点是否包含预定义节点呢:
一种暴力的做法是直接枚举所有给出的点, 求出高度进行判断。但是这样的复杂度多了nm的常数
考虑反过来, 对于每个预定义点, 把它的所有祖先的标记一下, 不难想到, 让节点编号每次除2就行。需要一个set维护, 同时还需要一个map来记录颜色。
总复杂度(O(nmlog(nm)))以及一个较大的常数
实现
赛时代码过丑抱歉
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <string>
#include <cassert>
#include <cstdlib>
#include <map>
#include <set>
#define int long long
using namespace std;
int read(){
int num=0, flag=1; char c=getchar();
while(!isdigit(c) && c!='-') c=getchar();
if(c=='-') flag=-1, c=getchar();
while(isdigit(c)) num=1ll*num*10+c-'0', c=getchar();
return num*flag;
}
int readc(){
char c=getchar();
while(c<'a' || c>'z') c=getchar();
return c;
}
void reads(){
char c=getchar();
while(c<'a' || c>'z') c=getchar();
while(c>='a' && c<='z') c=getchar();
}
int checkColor(int a, int b){
if(a > b) swap(a, b);
if(a == b) return 1;
if(a==4 && b==5) return 1;
if(a==2 && b==3) return 1;
if(a==0 && b==1) return 1;
return 0;
}
const int N = 3055, M = 5e6;
const int mod = 1e9+7;
int T, n, m;
int r[M][6], sz=0;
set<int> st;
map<int, int> col;
int get(char s){
if(s == 'w') return 0;
if(s == 'y') return 1;
if(s == 'g') return 2;
if(s == 'b') return 3;
if(s == 'r') return 4;
return 5;
}
int p[N], c;
int res[N][6];
int getlas(int id){
int num = 0;
while(id) num++, id>>=1;
return n-num+1;
}
int cnt = 0;
int check(int id){
return st.count(id);
}
int dp(int id, int h) {
if(h == 1){
int lres = (++sz)%M;
if(col.count(id)) {
for(int i=0; i<6; i++) r[lres][i] = 0;
r[lres][col[id]] = 1;
return lres;
}else{
for(int i=0; i<6; i++) r[lres][i] = 1;
return lres;
}
}
if(check(id)){
int r1=dp(id*2, h-1), r2=dp(id*2+1, h-1);
int lres = (++sz)%M;
if(col.count(id)) {
int i = col[id];
for(int j=0; j<6; j++) r[lres][j]=0;
for(int j=0; j<6; j++){
for(int k=0; k<6; k++){
if(checkColor(i, j) || checkColor(i, k)) continue;
r[lres][i] += r[r1][j]*r[r2][k]%mod;
r[lres][i] %= mod;
}
}
return lres;
}else{
for(int j=0; j<6; j++) r[lres][j]=0;
for(int i=0; i<6; i++){
for(int j=0; j<6; j++){
for(int k=0; k<6; k++){
if(checkColor(i, j) || checkColor(i, k)) continue;
r[lres][i] += r[r1][j]*r[r2][k]%mod;
r[lres][i] %= mod;
}
}
}
return lres;
}
}else{
int lres = (++sz)%M;
for(int i=0; i<6; i++){
r[lres][i] = res[h][i];
}
return lres;
}
}
signed main(){
n=read(), m=read();
for(int i=1; i<=m; i++){
p[i]=read();
c=get(readc()); reads();
int cur = p[i];
while(cur) {
if(!st.count(cur)) st.insert(cur);
cur /= 2;
}
col[p[i]] = c;
}
int num = 1;
for(int i=1; i<=n; i++){
for(int j=0; j<6; j++) res[i][j] = num;
num = (num*num%mod*16%mod);
}
int x = dp(1, n);
int ans = 0;
for(int i=0; i<6; i++) ans = (ans + r[x][i]) % mod;
cout << ans << endl;
return 0;
}