【P2476】着色方案(记忆化搜索+特殊的DP数组)
这个题代码难度几乎为0,然而思维难度对于蒟蒻来说简直是突破天际啊!首先我思考的是这个油漆的种类只有15种,是不是可以像一道叫做8数码难题的东西暴力15维数组呢。。计算发现不可以。。。。空间会直接让你学会做人,然后就是看到每种油漆最多有5桶,所以自然想到把这几桶分别有几种油漆作为状态转移,然后,蒟蒻就没有继续下去的思路了。。点开题解先看到这题与王八棋那道题的思路类似所以回忆了一下,有了一个大概的思路,就是按照这个方案记忆化搜索,然后根据乘法原理,直接乘上刚刚搜到的就可以了。没想到的是,一遍就过了。。。
#include<iostream> #include<cstdio> #include<cmath> #include<cstring> #include<algorithm> #define re register #define mo 1000000007 #define dp f[a][b][c][d][e][l] using namespace std; long long al[6],f[16][16][16][16][16][6],c,n,d,ans; inline long long dfs(int a,int b,int c,int d,int e,int l) { if(dp) return dp; if(a+b+c+d+e==0) return 1; if(a){dp+=(a-(l==2))*dfs(a-1,b,c,d,e,1);} if(b){dp+=(b-(l==3))*dfs(a+1,b-1,c,d,e,2);} if(c){dp+=(c-(l==4))*dfs(a,b+1,c-1,d,e,3);} if(d){dp+=(d-(l==5))*dfs(a,b,c+1,d-1,e,4);} if(e){dp+=e*dfs(a,b,c,d+1,e-1,5)%mo;} dp%=mo; return dp; } int main() { cin>>c; for(re int i=1;i<=c;i++) { cin>>d; al[d]++; } ans=dfs(al[1],al[2],al[3],al[4],al[5],0)%mo; cout<<ans; }
对于作者转载文章,欢迎继续转载。
对于作者原创文章,请注明出处之后转载。