代码源每日一题div1 Ayoub‘s function

Daimayuan Online Judge

思路：思维
我们从问题的反面考虑，就是要求所有的子串减去只含$0$的子串，那问题转化成如何让只含$0$的子串尽可能的少，结论是将$1$尽可能的均匀的分布在这个字符串中
显然我们有$m$个$1$就会分割出$m+1$个只含$0$的区间，那我们只要用总的子串个数减去这些只含$0$的区间的子串即可
一个长度为$l$的字符串的子串个数为：$l\ast (l+1)/2$
现在我们来证明一下这个结论，为了方便我们只考虑整除的情况，即这些$1$划分的$0$区间长度相同
我们定义字符串长度$n$，$1$的个数$m$，那么只含$0$的区间个数为$m+1$，这些区间的长度为$k=\frac{n-m}{m+1}$，那么只含$0$的子串个数$cnt=(m+1)\ast \frac{1}{2}\ast k\ast (k+1)$，现在我们将其中一个$1$向一侧移动一个位置，那么会导致两侧的两个$0$长度分别$-1$，$+1$，那么子串个数变为$cnt’=(m-1)\ast \frac{1}{2}\ast k\ast (k+1)+\frac{1}{2}\ast (k+1)\ast (k+2)+\frac{1}{2}\ast (k-1)\ast k$，做差比较：$cnt-cn{t}^{\prime }=\frac{1}{2}(2\ast k\ast (k+1)-(k+1)\ast (k+2)-k\ast (k-1))=-1<0$，因此$cnt<cn{t}^{\prime }$得证

Code：

#include <bits/stdc++.h>
using namespace std;
#define IOS ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
#define endl '\n';
typedef long long ll;
typedef pair<ll,ll> PII;
const int N=1e6+10;
ll n,m,t;
int main(){
	IOS;
	cin>>t;
	while(t--){
		cin>>n>>m;
		ll ans=n*(n+1)/2,x=(n-m)/(m+1),y=(n-m)%(m+1);
		if((n-m)%(m+1)==0) ans-=(m+1)*x*(x+1)/2;
		else ans-=((m+1-y)*x*(x+1)/2+y*(x+1)*(x+2)/2);
		cout<<ans<<endl;
	}
	return 0;
}