[CQOI2007]余数求和

问题：余数求和

题意：
给出正整数 $n$ 和 $k$，请计算 $$G(n,k)=\sum _{i=1}^{n}k\mathrm{m}\mathrm{o}\mathrm{d}i$$其中 $k\mathrm{m}\mathrm{o}\mathrm{d}i$ 表示 $k$ 除以 $i$ 的余数。
1≤n,k≤$10^9$。

例子：
G(10,5)=0+1+2+1+0+5+5+5+5+5=29

思路：
首先就是枚举呗。可以看到几个关键点：

• 核心就是k，k是不变的，而i从1循环到n。
• k对大于k的数取模，得数永远是k。

第一件事就是分类，把n分成小于等于k的部分和大于k的部分，时间复杂度是O(n)，会超时

#include<iostream>
#include<cstdio>
using namespace std;
typedef long long ll;
ll g(int n,int k){
    int m=min(n,k);
    ll res=m==k?1ll*k*(n-k):0;
    for(int i=1;i<=m;i++){
        res+=k%i;
    }
    return res;
}
int main(){
    int n,k;
    cin>>n>>k;
    cout<<g(n,k);
    return 0;
}

不出所料，果然超时。。。

考虑优化：

大于k的部分都O(1)了，肯定是不可能再优化的了；

看小于等于k的部分，由于k%i==k-i*k/i，${\sum }_{i=1}^{m}k\mathrm{m}\mathrm{o}\mathrm{d}i$可分解成k×m+${\sum }_{i=1}^{m}i\times k/i$ ，这样只需要算${\sum }_{i=1}^{m}i\times k/i$ 就行了

而对于${\sum }_{i=1}^{m}k/i$，我们知道可以用整除分块O($\sqrt{n}$)求出

ll f(int k,int m){
    ll res=0;
    for(int l=1,r;l<=m;l=r+1){//  这里把m/l看成整体，m/l相同的数组成的区间的个数有sqrt(n)个
        r=m/(m/l);//  算出m/l相同的数组成的区间的右边界
        res+=(r-l+1)*(m/l);//  所有m/l相同的数在一起被加起来
    }
    return res;
}

我们知道了${\sum }_{i=1}^{m}k/i$，也不难求出${\sum }_{i=1}^{m}i\times k/i$

ll f(int k,int m){
    ll res=0;
    for(int l=1,r;l<=m;l=r+1){//  还是把m/l看成整体
        r=m/(m/l);
        res+=(m/l)*(r-l+1)*(l+r)/2;//  所有m/l相同的数乘上各自的权值被加起来
    }
    return res;
}

这样${\sum }_{i=1}^{m}k\mathrm{m}\mathrm{o}\mathrm{d}i$ 就很好求了， $$G(n,k)=\sum _{i=1}^{n}k\mathrm{m}\mathrm{o}\mathrm{d}i$$ 也就很好求了

#include<iostream>
#include<cstdio>
using namespace std;
typedef long long ll;

// ll g(int n,int k){
//     int m=min(n,k);
//     ll res=m==k?1ll*k*(n-k):0;
//     for(int i=1;i<=m;i++){
//         res+=k%i;
//     }
//     return res;
// }

ll g(int n,int k){
    int u=min(n,k);
    ll res=1ll*k*u;
    for(int l=1,r;l<=u;l=r+1){
        r=min(k/(k/l),u);
        res-=1ll*(k/l)*(r-l+1)*(l+r)/2;
    }
    if(n>k){
        res+=1ll*k*(n-k);
    }
    return res;
}

int main(){
    int n,k;
    cin>>n>>k;
    cout<<g(n,k);
    return 0;
}

同样调用g函数，同样的1e9的数据，上面的那个需要10秒，下面的这个只需要1秒。
学会算法，走遍天下\dog