UVA 1363 - Joseph's Problem（整除分块）

题目链接 https://cn.vjudge.net/problem/UVA-1363

【题意】
输入正整数$n,k(1<=n,k<=10^9)$ 计算

$$\sum^{n}_{i=1}k \ mod \ i$$

【思路】
当$n>k$时，余数始终为$k$
当$n<=k$时，设

$$p =\lfloor \frac{k}{i} \rfloor , k \ mod \ i =k-pi$$ 如果有 $$\lfloor \frac{k}{i+1} \rfloor = p, k \ mod \ (i+1) = k-p(i+1)=k-pi-p=k \ mod \ i - p$$ 说明只要 $k$ 除以 $i$ 的整数部分相同，那么 $k \ mod \ i$ 就会形成一个等差数列，用一次求和公式即可

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

ll n,k;

int main(){
    while(scanf("%lld%lld",&n,&k)==2){
        ll ans=0;
        if(n>k){
            for(ll L=1,R;L<=k;L=R+1){
                R=k/(k/L);
                ans+=(k%L+k%R)*(R-L+1)/2LL;
            }
            ans+=(n-k)*k;
        }
        else{
            for(ll L=1,R;L<=n;L=R+1){
                R=k/(k/L);
                if(R<=n) ans+=(k%L+k%R)*(R-L+1)/2LL;
                else ans+=(k%L+k%n)*(n-L+1)/2LL;
            }
        }
        printf("%lld\n",ans);
    }
    return 0;
}