codeforces 1051d D. Bicolorings (dp)
阿新 • • 發佈:2018-12-11
思路: 因為每一行最多隻有2個,所以我們可以用01 分別表示白黑,那麼對於每一列 直接dp dp[ i ][ j ][ k ] 表示當前是第i 列,現在有j個聯通分量,k標誌最後一列的狀態,只有4種 分別為 00 01 10 11
程式碼:
#include<bits/stdc++.h> 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\n",ans); return 0; }