[BZOJ3560] DZY Loves Math V

传送门
啊终于有一道我会自己做的啦ヾ(◍°∇°◍)ノ゙

发现可以对每个质因子分别考虑贡献,最后乘起来。
对于一个质因子 (p) ,设它在 (i) 个数里出现了 (a_i) 次,令 (sum=prod (1+p^1+p^2+ cdots +p^{a_i}))
那么它的贡献就是 ((sum-1) cdot dfrac{p-1}{p}+1)

然后很有信仰地暴力分解一波质因子就过了呢ヾ(✿゚▽゚)ノ

#include<bits/stdc++.h>
#include<tr1/unordered_map>
#define LL long long
#define re register
#define fr(i,x,y) for(re int i=(x);i<=(y);i++)
#define rf(i,x,y) for(int i=(x);i>=(y);i--)
#define frl(i,x,y) for(int i=(x);i<(y);i++)
using namespace std;
using namespace tr1;
const int N=10000006;
const int INF=2147483647;
const int p=1e9+7;
int n;
LL s[N];

void read(int &x){
	char ch=getchar();x=0;
	for(;ch<'0'||ch>'9';ch=getchar());
	for(;ch>='0'&&ch<='9';ch=getchar()) x=(x<<3)+(x<<1)+ch-'0';
}

inline void Add(LL &x,LL y){
	x+=y;
	while(x<0) x+=p;
	while(x>=p) x-=p;
}

LL qpow(LL a,int n){
	LL ans=1;
	for(LL sum=a;n;n>>=1,sum=sum*sum%p) if (n&1) ans=ans*sum%p;
	return ans;
}

int main(){
	read(n);
	int x;
	fr(i,1,n){
		read(x);
		for(int j=2;j*j<=x;j++)
		 if (x%j==0){
		 	LL cnt=1,xx=1;
		 	while(x%j==0) x/=j,xx=xx*j%p,Add(cnt,xx);
		 	if (!s[j]) s[j]=cnt;
		 	 else s[j]=s[j]*cnt%p;
		 }
		if (x){
			if (!s[x]) s[x]=x+1;
			 else s[x]=s[x]*(x+1)%p;
		}
	}
	LL ans=1;
	frl(i,1,N)
	 if (s[i]){
	 	LL sum=(s[i]-1)*(i-1)%p;
	 	sum=sum*qpow(i,p-2)%p;
	 	ans=ans*(sum+1)%p;
	 }
	cout<<ans<<endl;
	return 0;
}
原文地址:https://www.cnblogs.com/ymzqwq/p/bzoj3560.html