LGP7704题解

来一个特别暴力的做法。

首先，如果删掉 \(x\) 和 \(y\) 的效果一定和删掉 \(xy\) 的效果相同，且代价一定不大于后者。

于是我们只删除质数，题目就变成了寻找 \(i!(1 \leq i \leq \max n)\) 中有多少个质数出现了奇数次。

给差分一下，变成求 \(i\) 的质因子分解。

有人可能会认为这里需要用到 PR，其实并不需要，因为所有的 \(i \leq \max n\)，所以只需要记录下每一个 \(i\) 最小的质因子，然后直接跳即可。

复杂度的话，\(i\) 的质因子总数一定不超过 \(\log_2 i\)（最坏情况是存在一个 \(k\) 使得 \(i=2^k\)），所以复杂度是 \(O(n\log \log n+T)\) 的，且常数较小

加上 快读&快输&加减法优化&整数除法转实数乘法&预处理逆元 后最大点才 349ms，轻松通过。

ps:这题思维难度和代码难度都不高，评绿吧。。。

#include<cstdlib>
#include<cstdio>
#include<cctype>
const int M=5e6+5,mod=998244353;
int T,n,k,top,p[M],ans[M],pri[M],pos[M];bool vis[M];
double inv[M];
inline int read(){
	int n=0;char s;
	while(!isdigit(s=getchar()));
	while(n=n*10+(s^48),isdigit(s=getchar()));
	return n;
}
inline void write(int n){
	char s[10];int top=0;
	while(s[++top]=n%10^48,n/=10);
	while(putchar(s[top]),--top);
}
inline int Add(const int&a,const int&b){
	return a+b>=mod?a+b-mod:a+b;
}
inline int Del(const int&a,const int&b){
	return b>a?a-b+mod:a-b;
}
inline int pow(int a,int b){
	int ans=1;
	for(;b;b>>=1,a=1ll*a*a%mod)if(b&1)ans=1ll*ans*a%mod;
	return ans;
}
inline void sieve(const int&M){
	register int i,j,x;
	for(i=2;i<=M;++i){
		ans[x=i]=ans[i-1];
		if(!pos[i]){
			pri[pos[i]=++top]=i;inv[top]=1./i+1e-15;
			ans[i]=Add(ans[i],p[top]=pow(i,k));vis[top]=1;
		}
		else{
			while(x^1){
				ans[i]=(vis[pos[x]]^=1)?Add(ans[i],p[pos[x]]):Del(ans[i],p[pos[x]]);
				x=x*inv[pos[x]];
			}
		}
		for(j=1;(x=i*pri[j])<=M;++j)if((pos[x]=j)==pos[i])break;
	}
}
signed main(){
	register int i;
	T=read();k=read();sieve(read());
	while(T--)write(ans[read()]),putchar(10);
}