[CodeForces 118D] Caesar's Legions （计数DP）

CodeForces - 118D
有 N1个步兵和 N2个骑兵排成一列
其中不能有连续的 K1个骑兵，连续的 K2个步兵
问有多少种排列方式

这题计数的我刚开始想得比较复杂
妄图用组合数和排列数给算出来
后来经菊苣提醒，发现有更无脑的做法

这种连续序列相关的，事实上可以用计数 DP来做
dp[i][j][k] 表示当前是第 i个士兵，其中用了 j个步兵
而 k如果是 0..K1则表示包括这位，之前有连续的 k个步兵
如果 k是 K1+1..K1+1+K2，则表示包括这位之前有连续的 k-(K1+1)个骑兵
这样就可以把用了几个步兵，几个骑兵，前面是连续的几个什么兵全表示出来
然后以第 i位，转移到第 i+1位
摆放步兵的时候，顺便转移到前面有连续的 0个骑兵的状态
同理骑兵，以方便转移

事实上有很多复杂的计数难以用组合排列，列公式表示出来
此时不妨使用 DP，将状态全都表示出来，设定好初值，拿去转移
这样减少思维难度，复杂度也很合理

#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <map>
#include <set>
#include <queue>
using namespace std;
typedef pair<int,int> Pii;
typedef long long LL;
typedef unsigned long long ULL;
#define MST(a,b) memset(a,b,sizeof(a))
#define CLR(a) MST(a,0)
#define Pow2(a) (a*a)

const int MOD=1e8;
int N1,N2,K1,K2;
int dp[210][110][30];

int main()
{
    scanf("%d%d%d%d", &N1, &N2, &K1, &K2);
    dp[1][1][1]=dp[1][1][K1+1]=1;
    dp[1][0][0]=dp[1][0][K1+2]=1;
    for(int i=1; i<N1+N2; i++)
    {
        for(int j=0; j<=min(i,N1); j++)
        {
            if(j<N1)
            {
                for(int k=0; k<=K1-1; k++)
                {
                    if(!dp[i][j][k]) continue;
                    dp[i+1][j+1][k+1]=((LL)dp[i+1][j+1][k+1]+dp[i][j][k])%MOD;
                    dp[i+1][j+1][K1+1]=((LL)dp[i+1][j+1][K1+1]+dp[i][j][k])%MOD;
                }
            }
            if((i-j)<N2)
            {
                for(int k=K1+1; k<=K1+1+K2-1; k++)
                {
                    if(!dp[i][j][k]) continue;
                    dp[i+1][j][k+1]=((LL)dp[i+1][j][k+1]+dp[i][j][k])%MOD;
                    dp[i+1][j][0]=((LL)dp[i+1][j][0]+dp[i][j][k])%MOD;
                }
            }
        }
    }
    int ans=0;
    for(int i=0; i<=K1; i++)
    {
        ans=((LL)ans+dp[N1+N2][N1][i])%MOD;
    }
    printf("%d\n", ans);
    return 0;
}