cf1051 D. Bicolorings

没想到是能开三维的

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const long long MOD=998244353;
ll dp[1010][2010][5];
int main()
{
    int n,kk;
    scanf("%d%d",&n,&kk);
    dp[1][1][0]=1;
    dp[1][2][1]=1;
    dp[1][2][2]=1;
    dp[1][1][3]=1;
    for(int i=2;i<=n;i++)
    {
        for(int j=1;j<=kk;j++)
        {
            dp[i][j][0]=(dp[i-1][j][0]+dp[i-1][j][1]+dp[i-1][j][2]+dp[i-1][j-1][3])%MOD;
              dp[i][j][1]=(dp[i-1][j-1][0]+dp[i-1][j][1]+dp[i-1][j-2][2]+dp[i-1][j-1][3])%MOD;
              dp[i][j][2]=(dp[i-1][j-1][0]+dp[i-1][j-2][1]+dp[i-1][j][2]+dp[i-1][j-1][3])%MOD;
            
            dp[i][j][3]=(dp[i-1][j-1][0]+dp[i-1][j][1]+dp[i-1][j][2]+dp[i-1][j][3])%MOD;
            //printf("i:%d j:%d :%d %d %d %d\n",i,j,dp[i][j][0],dp[i][j][1],dp[i][j][2],dp[i][j][3]);
        }
    }
    ll res=0;
    for(int i=0;i<=3;i++)
    {
        res+=dp[n][kk][i];
        res%=MOD;
    }
    printf("%lld\n",res);
}