CodeForces Hello 2019 1097D - Makoto and a Blackboard(积性函数)

首先设$E$是一阶期望，显然有下式成立,这说明$E$是积性函数.
$$E(n)=\frac{\sigma (n)}{d(n)}=\prod _{p^a\Vert n}\frac{1+p+\cdots +p^a}{a+1}=\prod _{p^a\Vert n}E(p^a)$$
其次假设$k-1$阶期望也是积性函数，那么$E_k$显然有下式成立:
$$E_k(n)=\frac{1}{d(n)}\sum _{t|n}{E}_{k-1}(t)=\prod _{p^a\Vert n}\frac{E_{k-1}(1)+E_{k-1}(p)+\cdots +E_{k-1}(p^a)}{a+1}=\prod _{p^a\Vert n}{E}_k(p^a)$$
所以$E_k$也是积性函数.
从而只用算每个素因子的贡献即可.
下面$E[i][k][a]$表示第$i$个素因子$p$的$k$阶期望$E_k[p^a]$,$pE$表示对应的前缀和.

#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
const long long mod=1e9+7;
long long qp(long long a,long long b){
	long long res=1;
	while(b){
		if(b&1)res=res*a%mod;
		a=a*a%mod;
		b>>=1;
	}
	return res;
}
long long p[20],pE[20][10005][70],E[20][10005][70],pc[20];
int main(){
	long long n;
	int k,c=0;
	scanf("%I64d%d",&n,&k);
	int sqrtn=(int)sqrt(n+1);
	for(int i=2;i<=sqrtn;i++){
		if(n%i==0){
			p[++c]=i;
			pc[c]=1;
			n/=i;
			while(n%i==0){n/=i;pc[c]++;}
		}
	}
	if(n){
		p[++c]=n;
		pc[c]=1;
	}
	long long res=1;
	for(int i=1;i<=c;i++){
		E[i][0][0]=pE[i][0][0]=1;
		for(int l=1;l<=pc[i];l++){
			E[i][0][l]=E[i][0][l-1]*p[i]%mod;
			pE[i][0][l]=(pE[i][0][l-1]+E[i][0][l])%mod;
		}
		for(int j=1;j<=k;j++){
			E[i][j][0]=pE[i][j][0]=1;
			for(int l=1;l<=pc[i];l++){
				E[i][j][l]=pE[i][j-1][l]*qp(l+1,mod-2)%mod;
				pE[i][j][l]=(pE[i][j][l-1]+E[i][j][l])%mod;
			}
		}
		res*=E[i][k][pc[i]];
		res%=mod;
	}
	printf("%I64d\n",res);
}