51nod 1769 Clarke and math2 线性筛+数学

题意

1 ≤ n ≤ 500000, 1 ≤ k ≤ 10^1000000

分析

刚看题目的时候被这个k的大小吓到了。。。
你考虑枚举每个f(i)对g(j)的贡献，那么它们的系数显然只跟j/i有关，设为val(j/i)。考虑如何求出val[x]。
从i变到j的过程我们可以看做变化k-1次，每一次都乘上一个数，只要满足每个数乘起来是x的因数即可。
考虑x的每种素因数对val[x]的贡献。不妨设该种素因数有s个。
那么我们可以选任意个出来，并将其分配到这k-1次变化当中，用隔板法来求，就相当于是$\sum\limits_{i=0}^sC_{i+k-2}^{k-2}=C_{k+x-1}^{k-1}$。
其中$C_{k+x-1}^{k-1}=\frac{(k+x-1)!}{(k-1)!*x!}=\frac{k*(k+1)*...*(x+k-1)}{x!}$。
那么val[x]就可以用线性筛来求了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=500005;
const int MOD=1000000007;

int n,tot,prime[N],low[N],num[N],val[N],ans[N],f[N],pow[N],ny[N],k;
bool not_prime[N];
char ch[1000005];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void get_prime(int n)
{
    for (int i=2;i<=n;i++)
    {
        if (!not_prime[i]) prime[++tot]=i,low[i]=i,num[i]=1;
        for (int j=1;j<=tot&&i*prime[j]<=n;j++)
        {
            not_prime[i*prime[j]]=1;
            if (i%prime[j]==0)
            {
                low[i*prime[j]]=low[i]*prime[j];
                num[i*prime[j]]=num[i]+1;
                break;
            }
            low[i*prime[j]]=prime[j];
            num[i*prime[j]]=1;
        }
    }
    val[1]=1;
    for (int i=2;i<=n;i++) val[i]=(LL)val[i/low[i]]*pow[num[i]]%MOD*ny[num[i]]%MOD;
}

int main()
{
    n=read();
    scanf("%s",ch+1);
    int len=strlen(ch+1);
    for (int i=1;i<=len;i++) k=((LL)k*10+ch[i]-'0')%MOD;
    for (int i=1;i<=n;i++) f[i]=read();
    ny[0]=ny[1]=pow[0]=1;pow[1]=k;
    for (int i=2;i<=n;i++) ny[i]=(LL)(MOD-MOD/i)*ny[MOD%i]%MOD,pow[i]=(LL)pow[i-1]*(k+i-1)%MOD;
    for (int i=2;i<=n;i++) ny[i]=(LL)ny[i-1]*ny[i]%MOD;
    get_prime(n);
    for (int i=1;i<=n;i++)
        for (int j=i;j<=n;j+=i)
            ans[j]+=(LL)f[i]*val[j/i]%MOD,ans[j]-=ans[j]>=MOD?MOD:0;
    for (int i=1;i<=n;i++) printf("%d ",ans[i]);
    return 0;
}