[莫比乌斯反演] 51nod1584. 加权约数和

杜教筛学傻了…一道反演题…我竟然差点推成杜教筛？？

令

$$A(n)=\sum_{i=1}^n \sigma(in)$$

这个东西推一下就变成

$$\sum_{t|n}\mu(t)tf({n\over t})g({n\over t})$$

$$f(n)=\sum_{d|n}d,g(n)=\sum_{i=1}^nf(n)$$

这个可$O(n\ln n)$ 推

那么答案就是

$$2\sum_{i=1}^n iA(i)-\sum_{i=1}^ni\sigma(i^2)$$

预处理后 $O(1)$ 输出就好了

最后还被卡常啊……
最后能线性筛的东西全改成线性筛才过……

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <ctime>

using namespace std;

typedef long long ll;

const int N=1000010,P=1e9+7;

int p[N],mu[N],f[N],g[N],h[N],ans[N],fac[N],w[N];

inline void Pre(const int n){
  mu[1]=1; f[1]=1;
  for(int i=2;i<=n;i++){
    if(!p[i])
      p[++*p]=i,mu[i]=-1,fac[i]=i,w[i]=(1LL*i*i+i+1)%P,f[i]=i+1;
    for(int j=1;j<=*p && 1LL*i*p[j]<=n;j++){
      p[i*p[j]]=1;
      if(i%p[j]){
    mu[i*p[j]]=-mu[i];
    fac[i*p[j]]=p[j];
    w[i*p[j]]=1LL*w[i]*w[p[j]]%P;
    f[i*p[j]]=1LL*f[i]*f[p[j]]%P;
      }
      else{
    fac[i*p[j]]=fac[i]*p[j];
    if(fac[i]==i){
      fac[i*p[j]]=fac[i]*p[j];
      w[i*p[j]]=(w[i]+1LL*fac[i]*fac[i]%P*p[j]+1LL*fac[i]*fac[i]%P*p[j]%P*p[j])%P;
      f[i*p[j]]=(f[i]+fac[i]*p[j])%P;
    }
    else{
      fac[i*p[j]]=fac[i]*p[j];
      w[i*p[j]]=1LL*w[i/fac[i]]*w[p[j]*fac[i]]%P;
      f[i*p[j]]=1LL*f[i/fac[i]]*f[p[j]*fac[i]]%P;
    }
    break;
      }
    }
  }
  for(int i=1;i<=n;i++) g[i]=(g[i-1]+f[i])%P;
  for(int i=1;i<=n;i++)
    for(int j=i;j<=n;j+=i)
      h[j]=(h[j]+1LL*mu[i]*i*f[j/i]%P*g[j/i])%P;
  for(int i=1;i<=n;i++)
    ans[i]=(ans[i-1]+1LL*i*h[i])%P;
  w[1]=1;
  for(int i=1;i<=n;i++) w[i]=(1LL*i*w[i]+w[i-1])%P;
  for(int i=1;i<=n;i++)
    ans[i]=(2LL*ans[i]-w[i])%P,ans[i]=(ans[i]+P)%P;
}

int main(){
  freopen("1.in","r",stdin);
  freopen("1.out","w",stdout);
  int t,n; scanf("%d",&t); Pre(1000000);
  for(int i=1;i<=t;i++)
    scanf("%d",&n),printf("Case #%d: %d\n",i,ans[n]);
  return 0;
}