CQOI2007 余数求和

P2261 [CQOI2007]余数求和

题目概述：给出正整数n和k，计算G(n, k)=k mod 1 + k mod 2 + k mod 3 + … + k mod n的值，其中k mod i表示k除以i的余数。
数据规模：n,k <= 10^9
这个题暴力算法只能过30%的点。
思路：显然，这是一道数论题，要注意数据规模虽然在int范围内，但是不能保证ans在int范围内，所以要开长整型格式。如果不开长整型，它的分数和暴力分是一样的。
然后模拟一遍这个过程：
G(10, 5)
=5 mod 1 + 5 mod 2 + 5 mod 3 + 5 mod 4 + 5 mod 5 + … + 5 mod 10
=0+1+2+1+0+5+…+5
容易发现，它具有一个性质：就是当m/i相同的几个数，它对应的余数是一个递减的等差数列，然后这个就可以变为一个O(N)的算法。
代码：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
long long i,j,m,n,a1,an,k,p;
long long ans;

int r()
{
    int ans=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')
            f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        ans*=10;
        ans+=ch-'0';
        ch=getchar();
    }
    return ans*f;
}

int main()
{
    n=r(),k=r();
    i=1;
    while(i<=n)
    {
        p=k/i;
        if(p==0)
        j=n;
        else
        j=min(k/p,n);
        a1=k%i,an=k%j;
        ans+=(j-i+1)*(a1+an)/2;//等差数列前n和
        i=j+1;
    } 
    cout<<ans;
    return 0;
}