Educational Codeforces Round 120 (Rated for Div. 2) D. Shuffle(组合数学+巧妙去重)

题目传送门

题目大意

给定一个01字符串和k,你可以选择包含k个1的连续子序列，打乱顺序，问该01字符串最多能有多少种。

解题思路

显然，当我们选择了一个长度为M的子串，且子串中包含K个1，那么该子串的排列方式可以通过组合数学的方式得出$C_M^K$ 。
朴素的想法，我们可以枚举每一种尽可能长的包含k个1的子序列，但是这之中包含了很多重复的。
如何去重便是解题关键。
我们可以看相邻的两个极长子序列：

其中红色的部分就是重复的，当我们用两段极长的排列数减去红色部分的排列数，我们就得到了一段新的不含重复的字符串，依次类推，人就是如何合并两段的问题。

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long 
const ll p = 998244353;
const int maxn = 5005;
int c[maxn][maxn];
void get_c()
{
	for (int i = 0; i < maxn; i++)
	{
		c[i][0] = 1;
		for (int j = 1; j <= i; j++)
		{
			c[i][j] = (c[i - 1][j] + c[i - 1][j - 1]) % p;
		}
	}
}
void solve()
{
	int n, k;
	cin >> n >> k;
	string s;
	cin >> s;
	vector<int>a; a.push_back(0);
	for (int i = 0; i < n; i++)
	{
		if (s[i] == '1')a.push_back(i + 1);
	}
	a.push_back(n + 1);
	ll res = 0;
	for (int i = k; i < a.size() - 1; i++)
	{
		//cout <<i<<" "<< res << endl;
		res = (res + c[a[i + 1] - a[i - k] - 1][k]) % p;
		if (i > k)res = ((res - c[a[i] - a[i - k] - 1][k - 1]) % p+p)%p;
	}
	if (k == 0||a.size()-2<k)cout << 1 << endl;
	else cout << res << endl;
}
int main()
{
	ios::sync_with_stdio(false);
	get_c();
	int t = 1;
	//cin >> t;
	while (t--)solve();
	return 0;
}