D. Bicolorings

題目連結：http://codeforces.com/contest/1051/problem/D 這道dp題分類討論好多啊 $dp[i][j][k]$表示前 $i$ 個格子，有$j$個連通塊，最後一列的狀態是$k$ 的方案數

#include"bits/stdc++.h"
#define out(x) cout<<#x<<"="<<x
#define C(n,m) (m>n?0:(long long)fac[(n)]*invf[(m)]%MOD*invf[(n)-(m)]%MOD)
using namespace std;
 

typedef long long LL;
const int maxn=1e3+5;
const int MOD=998244353;
LL dp[maxn][maxn<<1][4];
int main()
{
	dp[1][1][0]=1;
	dp[1][1][3]=1;
	dp[1][2][2]=1;
	dp[1][2][1]=1;
	for(int i=2; i<=1000; i++)
	{
		for(int j=1; j<=i*2; j++)
		{
			for(int k=0; k<=3; k++)
			{

				//連通塊數量不變
				dp[i][
 
j][k]+=dp[i-1][j][k];//跟i-1的末尾的連通塊狀態一樣
				if(k==0||k==3)
				{
					dp[i][j][k]+=dp[i-1][j][k^1];//[2,3]
					dp[i][j][k]+=dp[i-1][j][k^2];//[1,3]
				}

				//連通塊數量增加一個
				if(k==0||k==3)dp[i][j][k]+=dp[i-1][j-1][k^k];
				else
				{
					dp[i][j][k]+=dp[i-1][j-1][0];
					dp[i][j][k]+=dp[i-1][j-1][3];
				}
 

				
				//連通塊數量增加兩個
				if(k==1||k==2)
				{
					if(j>=2)dp[i][j][k]+=dp[i-1][j-2][k^3];
				}
				dp[i][j][k]%=MOD;
			}
		}
	}
	int N,K;
	while(cin>>N>>K)
	{
		LL ans=0;
		for(int sta=0; sta<=3; sta++)ans+=dp[N][K][sta];
		ans%=MOD;
		cout<<ans<<endl;
	}
}