1257: [CQOI2007]余数之和sum
如此一道水题,却被边界虐的很惨。
给定n和k,求sum = k mod 1 + k mod 2 + k mod 3 + … + k mod n(1<=n ,k<=10^9);
枚举商(n/i),之后分出商相同的若干个区间,注意到每个区间都是一个等差数列,分段求和即可;
对于商i
左端点 l = n/(i+1)+1;
右端点 r = min(m,n/i);
然后求出每段的总和 :(n%l+n%r)*(r-l+1)/2;
下一个 i = n / (n/(i+1));
然后累加即可;
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
using namespace std;
int main()
{
long long n,m;
scanf("%lld%lld",&m,&n);
long long res(0);
if(m>n) res+=(m-n)*n,m=n;
for(long long i = 1; i <= n;)
{
long long r = min(m,n/i);
long long l = (n/(i+1))+1;
if(l<=r) res += (n%l+n%r)*(r-l+1)/2;
if(i==n) break;
i = n/(n/(i+1));
}
cout<<res<<endl;
return 0;
}