[2017湖南集训7-7]第一题 DP

考虑只有一位和为10，枚举是哪两个数和为10，剩下的先按和为9用问号补齐，其余的问号可以任何组合9。
设dp[i][x]为匹配完第i种对(0-9,1-8,2-7,3-6,4-5五种i)，用了x个括号的方案数,T[x]表示在组合10和补齐9的过程中已经用了多少个问号充当x。于是我们枚举另用k个问号充当第i种对，那么dp[i][j]=sigma{dp[i-1][last]*C（nq-last，k/2+T[i]）*C（nq-last-k/2-T[i]，k/2+T[9-i]）}(k<=j-T[i]-T[9-i])，其中last=j-k-T[i]-T[9-i]即原先已用的问号数，nq为总问号数，C为组合数。
但是还有一种情况，就是两个0可一起填在末尾（前导），也就是可以同时削去两个0，如果枚举消去的0的个数又退化会O(n^3)，但是我们发现初始状态只有dp[0][0]=1，其余都为0，我们只用枚举0的个数和9的个数，看是否合法，先把dp[1][ ]搞出来即可。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#define ll long long
using namespace std;
const int mod=1e9+7;
int num[10],cnt[10],numq,t1[10],t2[10];
ll c[1010][1010],dp[6][1010],ans=0;
void solve(int nq)
{
    int d=nq;
    for(int i=1;i<=8;i++)
    {
        if(t1[i]<t1[9-i])t2[i]+=(t1[9-i]-t1[i]);
        d-=t2[i];
    }
    d-=t2[9];
    if(d<0) return ;
    memset(dp,0,sizeof(dp));
    dp[0][0]=1; 
    for(int i=0;i<=d;i++)
        for(int j=0;j<=d-i;j++)
            if(i+t1[0]>=j+t1[9]&&(i+t1[0]-j-t1[9])%2==0)    
                dp[1][i+j+t2[9]]=(dp[1][i+j+t2[9]]+c[nq][i]*c[nq-i][j+t2[9]]%mod)%mod;
    for(int i=1;i<=4;i++)
        {   
            for(int j=t2[i]+t2[9-i];j<=nq;j++)
                for(int k=0;k<=j-t2[i]-t2[9-i];k+=2)
                    {
                        int last=j-k-t2[i]-t2[9-i];
                        dp[i+1][j]=(dp[i+1][j]+dp[i][last]*c[nq-last][k/2+t2[i]]%mod*c[nq-last-k/2-t2[i]][k/2+t2[9-i]]%mod)%mod;
                    }   
        }

    ans=(ans+dp[5][nq])%mod;    
}
void step1(int nq)
{
    for(int i=1;i<=5;i++)
    {
        int del=0;
        memcpy(t1,num,sizeof(num));
        memcpy(t2,cnt,sizeof(cnt));
        if(t1[i]==0) t2[i]++;else t1[i]--;
        if(t1[10-i]==0) t2[10-i]++; else t1[10-i]--;
        solve(nq);  
    }   
}
int main()
{
    freopen("diyiti.in","r",stdin);
    freopen("diyiti.out","w",stdout);
    char s[100100];
    scanf("%s",s);
    for(int i=0;i<=strlen(s)-1;i++)
        if(s[i]=='?') numq++;
        else num[s[i]-'0']++;
    c[0][0]=1;
    for(int i=1;i<=numq;i++)
        for(int j=0;j<=numq;j++)
            c[i][j]=(c[i-1][j]+((j==0)?0:c[i-1][j-1]))%mod; 
    step1(numq);
    printf("%lld\n",ans);
    return 0;
}