古城宝藏题解 -- 求因数平方和

在一个古老的城市里，隐藏着一个关于宝藏的神秘谜题。传说中，城市的每一块基石都有其独特的力量值，而这种力量值由基石的特殊属性决定——即基石的所有正因子的平方和。

例如，基石编号3的力量值为 f(3) =  1*1+3*3。

城市的守护者提出了一个挑战：谁能计算出前n块基石的总力量值 g(n)，并且将这个总值对 10^9+7取模的结果告知他们，谁就能揭开古城宝藏的秘密。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll MOD=1e9+7;
ll n;
ll calc(ll x) {
    return (__int128)(x+1)*(2*x+1)/6%MOD;
}
ll check(ll t,ll x) {
    ll l=t,r=n,mid;
    while(l<=r) {
        mid=(l+r)/2;
        if(n/mid>=x) l=mid+1;
        else r=mid-1;
    }
    return r;
}
int main() {
    freopen("treasure.in","r",stdin);
    freopen("treasure.out","w",stdout);
    cin>>n;
    ll ans=0;
    for(ll i=1;i<=n;)
    {
        ll x=n/i,y=check(i,x);
        ans=(ans+(calc(y)-calc(i-1))*x)%MOD;
        i=y+1;
    }
    cout<<(ans+MOD)%MOD;
    return 0;
}