【SPOJ】DIVCNTK min_25筛

题目大意

　　给你 \(n,k\)，求
\[ S_k(n)=\sum_{i=1}^n\sigma_0(i^k) \]
　　对 \(2^{64}\) 取模。

题解

　　一个min_25筛模板题。

　　令 \(f(n)=\sigma_0(n^k)\)，那么 \(S_k(n)=\sum_{i=1}^nf(i)\)，而且
\[ \begin{cases} f(1)&=1\\ f(p)&=k+1\\ f(p^c)&=kc+1 \end{cases} \]
　　直接上min_25筛就好了。

　　时间复杂度：\(O(\frac{n^\frac{3}{4}}{\log n})\)

　　某些实现可能会有一点微小的细节。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<ctime>
#include<utility>
#include<functional>
#include<cmath>
#include<assert.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
void open(const char *s){
#ifndef ONLINE_JUDGE
    char str[100];sprintf(str,"%s.in",s);freopen(str,"r",stdin);sprintf(str,"%s.out",s);freopen(str,"w",stdout);
#endif
}
int rd(){int s=0,c,b=0;while(((c=getchar())<'0'||c>'9')&&c!='-');if(c=='-'){c=getchar();b=1;}do{s=s*10+c-'0';}while((c=getchar())>='0'&&c<='9');return b?-s:s;}
void put(int x){if(!x){putchar('0');return;}static int c[20];int t=0;while(x){c[++t]=x%10;x/=10;}while(t)putchar(c[t--]+'0');}
int upmin(int &a,int b){if(b<a){a=b;return 1;}return 0;}
int upmax(int &a,int b){if(b>a){a=b;return 1;}return 0;}
const int M=100010;
int pri[M],cnt,b[M];
ll f1[M],f2[M];
void init()
{
    for(int i=2;i<=100000;i++)
    {
        if(!b[i])
            pri[++cnt]=i;
        for(int j=1;j<=cnt&&i*pri[j]<=100000;j++)
        {
            b[i*pri[j]]=1;
            if(i%pri[j]==0)
                break;
        }
    }
    pri[cnt+1]=100001;
}
ll n,k;
int m;
ll dfs(ll x,int y)
{
    if(x<=1||x<pri[y])
        return 0;
    if(pri[y]>m)
        return (x<=m?f1[x]:f2[n/x])-f1[m];
    ll s=(x<=m?f1[x]:f2[n/x])-f1[pri[y]-1];
    for(int i=y;i<=cnt&&(ll)pri[i]*pri[i]<=x;i++)
    {
        ll x1=x/pri[i];
        for(int j=1;x1>=pri[i];j++,x1/=pri[i])
            s+=dfs(x1,i+1)*(j*k+1)+((j+1)*k+1);
    }
    return s;
}
void solve()
{
    scanf("%lld%lld",&n,&k);
    m=sqrt(n)+0.5;
    int mx=n/(m+1);
    for(int i=2;i<=m;i++)
        f1[i]=i-1;
    for(int i=1;i<=mx;i++)
        f2[i]=n/i-1;
    for(int i=1;i<=cnt&&pri[i]<=m;i++)
    {
        ll x=f1[pri[i]-1];
        int n1=min((ll)mx/pri[i],n/pri[i]/pri[i]);
        int n2=min((ll)mx,n/pri[i]/pri[i]);
        for(int j=1;j<=n1;j++)
            f2[j]-=f2[j*pri[i]]-x;
        for(int j=n1+1;j<=n2;j++)
            f2[j]-=f1[n/j/pri[i]]-x;
        for(int j=m;j>=(ll)pri[i]*pri[i];j--)
            f1[j]-=f1[j/pri[i]]-x;
    }
    for(int i=2;i<=m;i++)
        f1[i]*=k+1;
    for(int i=1;i<=mx;i++)
        f2[i]*=k+1;
    ll ans=dfs(n,1);
    ans++;
    printf("%llu\n",ans);
}
int main()
{
    open("divcntk");
    int t;
    scanf("%d",&t);
    init();
    while(t--)
        solve();
    return 0;
}