CF-Round #619-div2-C题

CF-Round #619-div2-C题

C. Ayoub’s function

好久没补题了=-=昨天忙着搭环境去了，可惜一直没有搞好=-=今天就来敲下代码把~
传送门

这算一个构造题把，但是数学成分比较重~

这道题我很喜欢，题目大概意思就是给你两个数，n和m，n是字符串的长度，m是字符串中1的个数，然后让你构造一个字符串，让含1的子字符串个数最多(注：单独的一个1也算）然后输出最多的子字符串的个数

解题思路：其实我们可以手动操作一下样例，可以发现，交替的01所获得的字符串可以获得最大的个数。所以我们需要把m个1给分开，相当于插入空位的操作，需要制造m + 1个位置，这样可以产生m个空位，让1去插入。即我们分成了m + 1个组，这时有n - m个0，平均每个组0的个数为k = (n - m) / (m + 1)（向下取整），然后多余还有kk = (n - m) % (m + 1)个0，然后我们把多余的0再次分配到m + 1个组中，我们就可以知道有kk个组中包含k + 1个0，然后m + 1 - kk个组中包含k个0，题目要求我们获得含有1的子字符串的个数，那我们为什么不反向思考一下呢，我们获得只含有0的字符串的个数，然后在在总字符串的个数里面剪掉这些，是不是就获得了我们的答案呢？所以我们求只含有0的字符串的个数，(m + 1 -kk ) * (k + 1) * k / 2 + kk * (k + 1) * (k + 2) / 2就是只含有0的字符串的个数的总和。
最后长度为n的子字符串的个数总和为（1 + n) * n / 2
减去就可以啦，
总式子为（1 + n) * n / 2 - (m + 1 -kk ) * (k + 1) * k / 2 + kk * (k + 1) * (k + 2) / 2
下面代码的式子就是上面的式子化简的结果啦~

上代码啦~

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int main()
{
	int t;
	cin >> t;
	while (t--)
	{
		ll n, m;
		cin >> n >> m;
		ll k = (n - m) / (m + 1);
		ll ans = n * (n + 1) / 2 - (m + 1) * k * (k + 1) / 2 - (k + 1) * ((n - m) % (m + 1));
		cout << ans << endl;
	}
	return 0;
} 

嘻嘻，完成啦~