1. 程式人生 > >bzoj 4407 於神之怒加強版 —— 反演+篩積性函數

bzoj 4407 於神之怒加強版 —— 反演+篩積性函數

targe ret scan 出現 space lse class cst www.

題目:https://www.lydsy.com/JudgeOnline/problem.php?id=4407

推導如這裏:https://www.cnblogs.com/clrs97/p/5191506.html

然後發現 \( F(D) \) 是一個積性函數,可以篩質數的同時篩出來;

首先,單個質數 \( p \) 時只有 \( d=1 \) 和 \( d=p \) 兩個因數,所以 \( F[p] = p^{k} - 1 \)

然後如果篩到互質的數,直接把 \( F() \) 相乘即可;

如果不互質,說明那個數已經有 \( p \) 這個質因子,考慮會在什麽地方出現;

觀察那個 \( \mu(d) \) 裏面的 \( d \),不會有兩個及以上相同的質因子,所以新加入的這個 \( p \) 不會在 \( d \) 中單獨出現了;

所以所有的 \( \mu(d) \) 外面那個 \( (\frac{D}{d})^{k} \) 裏面都會加入一個 \( p^{k} \),拿出來單獨乘上即可;

註意分塊的邊界是 \( min(n,m) \),因為枚舉的是 \( gcd \) ;

因為有負數,最後別忘了再模一下。

代碼如下:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
int const xn=5e6+5,mod=1e9+7;
int n,m,k,pri[xn],cnt,f[xn],s[xn];
bool vis[xn]; ll pw(ll a,int b) { ll ret=1; for(;b;b>>=1,a=(a*a)%mod)if(b&1)ret=(ret*a)%mod; return ret; } int upt(int x){while(x>=mod)x-=mod; while(x<0)x+=mod; return x;} void init() { int mx=5e6; f[1]=1; for(int i=2;i<=mx;i++) { if(!vis[i])pri[++cnt]=i,s[i]=pw(i,k),f[i]=upt(s[i]-1
); for(int j=1;j<=cnt&&(ll)i*pri[j]<=mx;j++) { vis[i*pri[j]]=1; if(i%pri[j])f[i*pri[j]]=(ll)f[i]*f[pri[j]]%mod; else {f[i*pri[j]]=(ll)f[i]*s[pri[j]]%mod; break;} } } for(int i=2;i<=mx;i++)f[i]=upt(f[i]+f[i-1]); } int main() { int T; scanf("%d%d",&T,&k); init(); while(T--) { scanf("%d%d",&n,&m); int mn=min(n,m),ans=0;// for(int i=1,j;i<=mn;i=j+1) { j=min(n/(n/i),m/(m/i)); ans=(ans+((ll)f[j]-f[i-1])*(n/i)%mod*(m/i))%mod; } printf("%d\n",upt(ans));//- } return 0; }

bzoj 4407 於神之怒加強版 —— 反演+篩積性函數