[DP] Atcoder AGC013D. Piling Up

假设刚开始有 $x$ 个红色， $n-x$ 个蓝色

一个合法的数列有多个初始的 $x$ 值，对于最小的 $x$ 值，在操作的过程中，$x$ 会变成0

那么 $f_{i,j,0..1}$ 表示操作了 $i$ 次，当前 $x=j$ ，$x$ 是否变成过0

答案就是 $\sum f_{m,i,1}$

这样就可以不重复了

#include <cstdio>
#include <iostream>
#include <algorithm>

using namespace std;

const int P=1e9+7;

int f[3010][3010][2];

inline void add(int &x,int y){ (x+=y)%=P; }

int main(){
    int n,m;
    cin>>n>>m;
    f[0][0][1]=1;
    for(int i=1;i<=n;i++) f[0][i][0]=1;
    for(int i=0;i<m;i++)
        for(int j=0;j<=n;j++)
            for(int k=0;k<=1;k++){
                if(j) add(f[i+1][j-1][k|!(j-1)],f[i][j][k]);
                if(n-j) add(f[i+1][j+1][k],f[i][j][k]);
                if(j) add(f[i+1][j][k|(j==1)],f[i][j][k]);
                if(n-j) add(f[i+1][j][k],f[i][j][k]);
            }
    int ans=0;
    for(int i=0;i<=n;i++) add(ans,f[m][i][1]);
    cout<<ans<<endl;
    return 0;
}