比较容易看出来是DP,但是如果我们记录每一种颜色还剩多少种的话,消耗的转移的时间复杂度5^15,但是我们考虑到每一种颜色,如果数量相同的话,其实是等效的,所以我们用w[a][b][c][d][e][last]记录还剩下1,2,3,4,5次使用次数的颜色的数量为a,b,c,d,e,上一次我们那的是使用次数剩余为last的颜色,那么这次在剩余last-1次的颜色中,我们只能少拿一次,因为不能拿相同的,这样转移就很容易的表示出来了。
我把存状态的数组都开到20就在0mstle了,改到16就好了,求指导。。。
/************************************************************** Problem: 1079 User: BLADEVIL Language: C++ Result: Accepted Time:80 ms Memory:131876 kb ****************************************************************/ //By BLADEVIL #include <cstdio> #define LL long long #define d39 1000000007 using namespace std; int n,x; int sum[20]; LL w[16][16][16][16][16][16]; LL calc(int a,int b,int c,int d,int e,int last) { if (a+b+c+d+e==0) return w[a][b][c][d][e][last]=1; if (w[a][b][c][d][e][last]) return w[a][b][c][d][e][last]; LL sum=0ll; if (a) (sum+=(a-(last==2))*calc(a-1,b,c,d,e,1))%=d39; if (b) (sum+=(b-(last==3))*calc(a+1,b-1,c,d,e,2))%=d39; if (c) (sum+=(c-(last==4))*calc(a,b+1,c-1,d,e,3))%=d39; if (d) (sum+=(d-(last==5))*calc(a,b,c+1,d-1,e,4))%=d39; if (e) (sum+=e*calc(a,b,c,d+1,e-1,5))%=d39; return w[a][b][c][d][e][last]=sum; } int main(){ scanf("%d",&n); while (n--) scanf("%d",&x),sum[x]++; LL ans=calc(sum[1],sum[2],sum[3],sum[4],sum[5],0); printf("%lld ",ans); return 0; }