C. Ayoub‘s function(思维+组合数学取补集)
作者:互联网
https://codeforces.com/problemset/problem/1301/C
题意:给出一个字符串中1的个数,和它的长度,其余都是0,然后构造一个字符串,使得f(s)最大,f(s)表示有多少个点对,使得这个点对的区间中至少包含一个1。
思路:取反,考虑只有0串的情况,然后拿总数-只有0串的情况。
1有m个,那么0有n-m个。于是要使1多,那么0串总数要尽量小。00的0串总数是3,而010的0串总数是2.所以构造的思路就是均摊,将0尽量插到1的间隔之间平均分配。
#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=1e5;
typedef long long LL;
inline LL read(){LL x=0,f=1;char ch=getchar(); while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
int main(void)
{
cin.tie(0);std::ios::sync_with_stdio(false);
LL t;cin>>t;
while(t--){
LL n,m;cin>>n>>m;
LL ans=n*(n+1)/2;
LL num0=n-m;
LL k=num0/(m+1);
LL mod=num0%(m+1);
LL sum1=(m+1-mod)*(k*(k+1)/2);
LL sum2=mod*((k+1)*(k+1+1)/2);
cout<<ans-(sum1+sum2)<<"\n";
}
return 0;
}
标签:function,总数,num0,Ayoub,LL,cout,取补集,include,mod 来源: https://blog.csdn.net/zstuyyyyccccbbbb/article/details/113802574