打表可以发现an是n*(n+1)
进而推出求和公式
容斥一下,代码如下
#include<cstdio>
#include<cstring>
using namespace std;
const long long mod=1e9+7;
int prime[33],pn;
long long n,m;
long long q_pow(long long a,long long b)
{
long long ans=1;
while(b){
if(b&1) ans=(ans*a)%mod;
a=(a*a)%mod;
b>>=1;
}
return ans;
}
long long sum(long long s,long long n)
{
long long temp1=(s*s)%mod,temp2=s;
temp1=(temp1*n)%mod;
temp1=(temp1*(n+1))%mod;
temp1=(temp1*(2*n+1))%mod;
temp1=(temp1*q_pow(6,mod-2))%mod;
temp2=(temp2*n)%mod;
temp2=(temp2*(1+n))%mod;
temp2=(temp2*q_pow(2,mod-2))%mod;
long long ans=(temp1+temp2)%mod;
return ans;
}
long long calc(long long n){
long long res=0;
for(int i=1; i<(1<<pn); ++i){
long long tmp=1,cnt=0;
for(int j=0; j<pn; ++j){
if(((i>>j)&1)==0) continue;
++cnt;
tmp*=prime[j];
}
long long ans=sum(tmp,n/tmp);
if(cnt&1) res=(res+ans)%mod;
else res=(res-ans+mod)%mod;
}
return res;
}
int main(){
while(~scanf("%lld%lld",&n,&m)) {
pn=0;
for(int i=2; i*i<=m; ++i){
if(m%i) continue;
while(m%i==0) m/=i;
prime[pn++]=i;
}
if(m!=1) prime[pn++]=m;
long long ans=sum(1,n);
// printf("%lld\n",ans);
ans=(ans-calc(n)+mod)%mod;
printf("%lld\n",ans);
}
}