H.Dividing(整除分块)

H.Dividing(整除分块)

思路：整除分块.

显然题意可以转化为：求满足$n\mathrm{m}\mathrm{o}\mathrm{d}k=0$或$1$的对数。

$n\mathrm{m}\mathrm{o}\mathrm{d}k=1\iff (n-1)\mathrm{m}\mathrm{o}\mathrm{d}k=0$

对于特殊情况：

当$k=1$时有$n$个，$n=1,k>1$时有$k-1$个。

然后我们只需计算出：$\sum _{i=2}^k\lfloor \frac{n}{i}\rfloor +\sum _{i=2}^k\lfloor \frac{n-1}{i}\rfloor$

因此可以用整除分块解决。

即：答案为：$n+k-1+\sum _{i=2}^k\lfloor \frac{n}{i}\rfloor +\sum _{i=2}^k\lfloor \frac{n-1}{i}\rfloor$

时间复杂度：$O(\sqrt{k})$

#include 
using namespace std;
typedef long long ll;
const double eps=1e-8;
const int mod=1e9+7;
ll n,k,ans=0;
void calc(ll n,ll k){
    for(ll l=2,r;l<=min(n,k);l=r+1){
        r=min(n/(n/l),k);
        ans=(ans+((r-l+1)%mod)*((n/l)%mod)%mod)%mod;
    }
}
int main(){
    scanf("%lld%lld",&n,&k);
    ans=(n+k-1)%mod;
    calc(n,k);
    calc(n-1,k);
    printf("%lld\n",ans);
    return 0;
}