【51Nod1769】Clarke and math2(数论,组合数学)
作者:互联网
【51Nod1769】Clarke and math2(数论,组合数学)
题面
题解
考虑枚举一个\(i_k\),枚举一个\(i\),怎么计算\(i_k\)对\(i\)的贡献。
把\(\frac{i}{i_k}\)拆掉,维护一个长度为\(k\)的数组,表示\(\frac{i_{k-1}}{i_{k}}\),对于每一个质因子,假设其出现次数为\(a\),那么就是把\(a\)个元素放进\(k\)个盒子里,盒子可以空,这个的方案数是\({a+k-1\choose k-1}={a+k-1\choose a}\),不难发现\(a\)很小,所以可以直接\(O(a)\)暴力算。
于是我们提前对于每一个\(a\)算出方案数,然后提前把质因数分解好,就可以做到\(O(nlogn)\)了。
#include<iostream>
#include<cstdio>
using namespace std;
#define MOD 1000000007
#define MAX 500500
inline int read()
{
int x=0;bool t=false;char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')t=true,ch=getchar();
while(ch<='9'&&ch>='0')x=(x*10ll+ch-48)%MOD,ch=getchar();
return t?-x:x;
}
int n,K,C[25],f[MAX],g[MAX],x[MAX],v[MAX],inv[25];
int main()
{
n=read();K=read();
for(int i=1;i<=n;++i)f[i]=read();
inv[0]=inv[1]=1;for(int i=2;i<=20;++i)inv[i]=1ll*inv[MOD%i]*(MOD-MOD/i)%MOD;
for(int i=1;i<=20;++i)inv[i]=1ll*inv[i-1]*inv[i]%MOD;
for(int a=0;a<=20;++a)
{
int nw=(a+K-1)%MOD;C[a]=inv[a];
for(int i=0;i<a;++i)C[a]=1ll*C[a]*(nw-i+MOD)%MOD;
}
for(int i=1;i<=n;++i)x[i]=i,v[i]=1;
for(int i=2;i<=n;++i)
if(x[i]!=1)
for(int j=i;j<=n;j+=i)
{
int a=0;
while(x[j]%i==0)++a,x[j]/=i;
v[j]=1ll*v[j]*C[a]%MOD;
}
for(int i=1;i<=n;++i)
for(int j=i;j<=n;j+=i)
g[j]=(g[j]+1ll*f[i]*v[j/i])%MOD;
for(int i=1;i<=n;++i)printf("%d ",g[i]);
puts("");return 0;
}
标签:ch,int,MAX,Clarke,51Nod1769,read,math2,getchar 来源: https://www.cnblogs.com/cjyyb/p/11139358.html