bzoj 1079 DP

  比较容易看出来是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\n",ans);

    return 0;

}       

 

你可能感兴趣的:(ZOJ)