牛客网练习赛27 E 欧拉（积性函数）

题目链接:https://www.nowcoder.com/acm/contest/188/E

#include<bits/stdc++.h>
using namespace std;

#define debug puts("YES");
#define rep(x,y,z) for(int (x)=(y);(x)<(z);(x)++)
#define read(x,y) scanf("%d%d",&x,&y)
#define ll long long

#define lrt int l,int r,int rt
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define root l,r,rt
const int  maxn =5e6+5;
const int mod=998244353;
const int ub=1e6;
ll powmod(ll x,ll y){ll t; for(t=1;y;y>>=1,x=x*x%mod) if(y&1) t=t*x%mod; return t;}
ll gcd(ll x,ll y){return y?gcd(y,x%y):x;}
/*
组合性质推式子，
假设我们已经知道了F(n),
如果d和n互质，F(n*d)=d^k*F(n)-F(n),
因为莫比乌斯函数自带组合性质，
假如miu函数选了d,则不难看出贡献是-F(n),
如果没选则直接在n^k上观察到需要扩大d^k。
如果不互质，同理，就是直接乘上d^k
*/
int prim[maxn/10],tot=0,kk;
int vis[maxn];
ll F[maxn];
void sieve()
{
    F[1]=1;
    for(int i=2;i<maxn;i++)
    {
        if(vis[i]==0) prim[tot++]=i,F[i]=(powmod(i,kk)-1+mod)%mod;
        for(int j=0;j<tot;j++)
        {
            if(1LL*i*prim[j]>=maxn) break;
            int k=i*prim[j];
            vis[k]=1;
            F[k]=F[i]*powmod(prim[j],kk)%mod;
            if(i%prim[j])  F[k]=(F[k]-F[i]+mod)%mod;
            else  break;
        }
    }
}
ll n,m;
int main()
{
    scanf("%lld%lld",&m,&kk);
    sieve();
    for(int i=0;i<m;i++)
    {
        scanf("%lld",&n);
        printf("%lld\n",F[n]);
    }
}