题目链接:Problem - 118D - Codeforces
题意:有个n1个步兵,n2个骑兵,求出一个满足连续步兵个数不超过m1且连续骑兵个数不超过m2的排列的方案数。
分析:
考虑dp,简单观察可以发现转移过程中需要知道当前步兵骑兵数量以及末尾序列是步兵还是骑兵
设f[i][j][k(0, 1)]表示步兵为i个骑兵为j个且以(步兵,骑兵)结尾的序列方案数
考虑f[i][j][0], 以步兵结尾,可以根据末尾步兵有几个划分状态,最多不超过min(i, m1)
方程f[i][j][0] += f[i - k][j][1], 连续步兵的最前面是骑兵,f[i][j][1]类似。
代码:
#include <bits/stdc++.h>
#define ios ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
#define x first
#define y second
#define endl '\n'
#define int long long
using namespace std;
typedef pair<int, int> PII;
const int N = 1e6 + 10;
const int M = 2e5 + 10;
const int INF = 0x3f3f3f3f;
const double INFF = 0x7f7f7f7f7f7f7f7f;
const int mod = 1e8;
int n1, n2, m1, m2;
int f[105][105][2];
//考虑i个步兵j个骑兵以步兵或骑兵结尾的方案数
signed main()
{
cin >> n1 >> n2 >> m1 >> m2;
f[0][0][0] = f[0][0][1] = 1;
for (int i = 0; i <= n1; i++)
{
for (int j = 0; j <= n2; j++)
{
int x = min({i, m1});
int y = min({j, m2});
int &a = f[i][j][0], &b = f[i][j][1];
for (int k = i - 1, p = 1; p <= x; p++, k--)
{
a = (a + f[k][j][1]) % mod;
}
for (int k = j - 1, p = 1; p <= y; p++, k--)
{
b = (b + f[i][k][0]) % mod;
}
}
}
cout << (f[n1][n2][1] + f[n1][n2][0]) % mod;
return 0;
}