LOJ#6053. 简单的函数

II.LOJ#6053. 简单的函数

重申一下min25筛应用的条件：

• 是积性函数。
• 质数处取值是低阶多项式。
• 质数次幂处取值可以快速求出。

满足以上三点的任意函数均可以min25筛。

现在看到这题。乍一看 $p\operatorname{xor}a$ 这种东西看上去一脸非多项式的样子；但是因为我们只关心质数处取值是否是多项式，因为除了 $2$ 以外的质数均为奇，且质数处的 $a$ 必为 $1$，则除了 $f(2)=3$ 以外，其余的 $f(p)=p-1$，是低阶多项式。故我们只需特判掉 $n\geq 2$ 时手动给筛出来的结果加上 $2$ 即可。

代码：

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
typedef long long ll;
ll n;
int m,pri[1001000],sp[2][1001000],g[2][1001000];
void sieve(){
	for(int i=2;i<=m;i++){
		if(!pri[i])pri[++pri[0]]=i,sp[0][pri[0]]=sp[0][pri[0]-1]+1,sp[1][pri[0]]=(sp[1][pri[0]-1]+i)%mod;
		for(int j=1;j<=pri[0]&&i*pri[j]<=m;j++){
			pri[i*pri[j]]=true;
			if(!(i%pri[j]))break;
		}
	}
}
ll kth[1001000];
int sml[1001000],lar[1001000],tot;
int H(ll x,int y){
	if(pri[y]>=x)return 0;
	int X=(x<=m?sml[x]:lar[n/x]);
	int ret=(2ll*mod+g[1][X]-g[0][X]-(sp[1][y]-sp[0][y]))%mod;
	for(int i=y+1;i<=pri[0]&&1ll*pri[i]*pri[i]<=x;i++){
		ll pa=pri[i];
		for(int a=1;pa<=x;a++,pa*=pri[i])(ret+=1ll*(pri[i]^a)*(H(x/pa,i)+(a!=1))%mod)%=mod;
	}
	return ret;
}
int main(){
	scanf("%lld",&n),m=sqrt(n),sieve();
//	for(int i=1;i<=pri[0];i++)printf("%d ",pri[i]);puts("");
//	for(int i=1;i<=pri[0];i++)printf("%d ",sp[0][i]);puts("");
//	for(int i=1;i<=pri[0];i++)printf("%d ",sp[1][i]);puts("");
	for(ll i=1;i<=n;i=n/(n/i)+1){
		kth[++tot]=n/i;
		int I=kth[tot]%mod;
		g[0][tot]=(I+mod-1)%mod,g[1][tot]=(1ll*I*(I+1)/2+mod-1)%mod;
		if(kth[tot]<=m)sml[kth[tot]]=tot;else lar[n/kth[tot]]=tot;
	}
//	for(int i=1;i<=tot;i++)printf("%d ",kth[i]);puts("");
	for(int i=1;i<=pri[0];i++)for(int j=1;j<=tot&&1ll*pri[i]*pri[i]<=kth[j];j++){
		ll nexi=kth[j]/pri[i];int I=(nexi<=m?sml[nexi]:lar[n/nexi]);
		(g[0][j]+=(mod-g[0][I]+sp[0][i-1])%mod)%=mod;
		(g[1][j]+=mod-1ll*pri[i]*(g[1][I]-sp[1][i-1]+mod)%mod)%=mod;
	}
	printf("%d\n",H(n,0)+1+2*(n>=2));
	return 0;
}