这场队友表现神勇,我也有所进步。(其实运气好,正好我和队友都会一点数论,前三条都是数学类型的)
其中有一条是整除分块的体型,今天的博客重点说一下整除分块的知识和重写H。
整除分块
求解问题:
找一个低复杂度的算法,即整除分块。
不考虑推导过程的话,引用某大佬的:
代码也是朴素又枯燥:
for(int l=1,r;l<=n;l=r+1)
{
r=n/(n/l);
ans+=(r-l+1)*(n/l);
}
H
比赛时我通过打表找规律,发现了如下规律:
当n%k=0或者n%k=1时满足题意
这与题解中的相符,即
然而数据量之大导致了我们不可能暴力枚举,队友说要用到整除分块的思想,让我们看看题解怎么说:
直接分成三部分求解
具体看代码如下:
#include<bits/stdc++.h>
using namespace std;
long long mod=1e9+7;
int main()
{
long long n,k;
scanf("%lld%lld",&n,&k);
long long ans=0;
for(long long l=2,r;l<=n&&l<=k;l=r+1)
{
r=min(n/(n/l),k);
ans=ans+(r-l+1)%mod*(n/l)%mod;
ans%=mod;
}
for(long long l=2,r;l<=n-1&&l<=k;l=r+1)
{
r=min((n-1)/((n-1)/l),k);
ans=ans+(r-l+1)%mod*((n-1)/l)%mod;
ans%=mod;
}
ans+=n+k-1;
ans%=mod;
printf("%lld",ans);
}
注意要开到long long,没开long long判题给出的是TLE,导致我两次都没发现这边的问题。
下面再补充一题队友推荐给我的整除分块的题:
U87406 整除分块
你懂的,整除分块+公式(不会吧不会吧,不会真的有人不知道平方数之和的公式吧)
然后
#include<bits/stdc++.h>
using namespace std;
const long long mod=1e9+7;
long long qpow(long long base,long long x)
{
long long ans=1;
while(x)
{
if(x&1)ans=ans*base%mod;
base=base*base%mod;
x>>=1;
}
return ans;
}
long long imp;
long long ans(int k,int n)
{
k=k-1;
//long long d=(((n*(n+1)%mod)%mod*(2*n+1)%mod)*imp)%mod-(((k*(k+1))%mod*(2*k+1)%mod)*imp)%mod;
long long d=n%mod*(n+1)%mod*(2*n+1)%mod*imp%mod-k%mod*(k+1)%mod*(2*k+1)%mod*imp%mod;
if(d<0)d+=mod;
return d;
}
int main()
{
int t;
scanf("%d",&t);
imp=qpow(6,mod-2);
while(t--)
{
int n;
scanf("%d",&n);
long long d=0;
for(int l=1,r;l<=n;l=r+1)
{
r=n/(n/l);
d=(d+(n/l)*ans(l,r))%mod;
}
printf("%lld\n",d);
}
}
P.S.卡了我一个下午的不是整除分块,也不是公式,而是取模
这个到学到取模的时候再重点关注一下。