【AGC013D】Pilling Up【动态规划】

首先我们很容易想到一个dp。
让f[i][j]f[i][j]表示ii次操作后有jj个红球的拿球的方案数。
易得转移
若j>0j>0，
f[i][j]−>f[i+1][j−1]f[i][j]−>f[i+1][j−1]
f[i][j]−>f[i+1][j]f[i][j]−>f[i+1][j]
若j<nj<n
f[i][j]−>f[i+1][j+1]f[i][j]−>f[i+1][j+1]
f[i][j]−>f[i+1][j]f[i][j]−>f[i+1][j]
但是这样子我们会算重复，因为询问的是颜色序列的方案数，不是拿球的方案数。
观察发现答案为共nn个球的答案减去共n−1n−1个球的答案，于是好像没了？
怎么证？我也不知道QAQ

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int N=3005,mod=1000000007;
int n,m,f[N][N*2];
int calc(int n){
    memset(f,0,sizeof(f));
    for(int i=0;i<=n;i++){
        f[0][i]=1;
    }
    for(int i=0;i<m;i++){
        for(int j=0;j<=n;j++){
            if(j>0){
                f[i+1][j-1]=(f[i+1][j-1]+f[i][j])%mod;
                f[i+1][j]=(f[i+1][j]+f[i][j])%mod;
            }
            if(j<n){
                f[i+1][j+1]=(f[i+1][j+1]+f[i][j])%mod;
                f[i+1][j]=(f[i+1][j]+f[i][j])%mod;
            }
        }
    }
    int ans=0;
    for(int i=0;i<=n;i++){
        ans=(ans+f[m][i])%mod;
    }
    return ans;
}
int main(){
    scanf("%d%d",&n,&m);
    printf("%d\n",(calc(n)-calc(n-1)+mod)%mod);
    return 0;
}

转载于:https://www.cnblogs.com/2016gdgzoi471/p/9476833.html