codeforces 1051 d D.Bicolorigs(dp)

リンク: http://codeforces.com/contest/1051/problem/D
考え方:行ごとに最大2つしかないので、01でそれぞれ白黒を表してもいいです.そうすると、各列に対して直接dp dp[i][j]、[k]は現在i列目を表しています.現在はj個の連結成分があります.kマークの最後の列の状態は4種類だけです.  それぞれ00-01 10 11です.
コード:

#include

using namespace std;
typedef long long ll;
const ll mod=998244353;
int n,K;

ll dp[1005][2005][4];

int main()
{
    cin>>n>>K;
    dp[1][1][0]=1;
    dp[1][2][1]=1;
    dp[1][2][2]=1;
    dp[1][1][3]=1;

    for(int i=1;i<=n;i++){
        for(int j=1;j<=2*n;j++){
            for(int k=0;k<=3;k++) dp[i][j][k]%=mod;
            dp[i+1][j][0]+=dp[i][j][0];
            dp[i+1][j+1][1]+=dp[i][j][0];
            dp[i+1][j+1][2]+=dp[i][j][0];
            dp[i+1][j+1][3]+=dp[i][j][0];

            dp[i+1][j][0]+=dp[i][j][1];
            dp[i+1][j][1]+=dp[i][j][1];
            dp[i+1][j+2][2]+=dp[i][j][1];
            dp[i+1][j][3]+=dp[i][j][1];

            dp[i+1][j][0]+=dp[i][j][2];
            dp[i+1][j+2][1]+=dp[i][j][2];
            dp[i+1][j][2]+=dp[i][j][2];
            dp[i+1][j][3]+=dp[i][j][2];

            dp[i+1][j+1][0]+=dp[i][j][3];
            dp[i+1][j+1][1]+=dp[i][j][3];
            dp[i+1][j+1][2]+=dp[i][j][3];
            dp[i+1][j][3]+=dp[i][j][3];

        }
    }

    ll ans=0;
    for(int i=0;i<=3;i++){
        ans+=dp[n][K][i];
        ans%=mod;
    }

    printf("%lld
",ans);

    return 0;
}