【HDU6417】Rikka with APSP（min_25筛）

传送门

---

题解：

考虑怎么算出 $a$ 到 $b$ 的最短路。

设 $S_{a,b}$ 表示 $a$ 和 $b$ 的唯一表示中指数奇偶性不一样的质数集合。

则 $$d_{a,b}=\{\begin{array}{rlr}1& & |S_{a,b}|=0\\ \prod _{p\in S_{a,b}}p& & |S_{a,b}|>0\end{array}$$

设 $di{s}_{a,b}$ 表示 $a,b$之间的最短路，容易发现：

$$di{s}_{a,b}=\{\begin{array}{rlr}1& & |S_{a,b}|=0\\ \sum _{p\in S_{a,b}}p& & |S_{a,b}|>0\end{array}$$

于是很显然地，我们需要分两部分计算。

先考虑 $|S_{a,b}|=0$ 的部分，也就是说 $a,b$ 所有质因子指数的奇偶性都相等。

把指数为奇数的质因子拿出来，设这些质因子乘积为 $c$ ，那么 $a/c,b/c$ 就是两个不同的完全平方数，其平方根在 $[1,\lfloor \sqrt{\frac{n}{c}}\rfloor ]$ 当中。而所有合法的 $c$ 其实就是所有无平方因子数。

这一部分的答案很容易写出来，就是 $$\sum _{i=1}^n{\mu }^2(i)\left(\genfrac{}{}{0ex}{}{\lfloor \sqrt{\frac{n}{i}}\rfloor }{2}\right)$$

后面可以整除分块，所以考虑处理前面的 ${\mu }^2$ 前缀和：

$$\sum _{i=1}^n{\mu }^2(i)=\sum _{i=1}^n\sum _{d^2\mid i}\mu (d)=\sum _{d=1}^{\sqrt{n}}\mu (d)\lfloor \frac{n}{d^2}\rfloor$$

线筛预处理一部分，后面的暴力计算，复杂度可以做到杜教筛那样。

第二部分由于是求和形式，所以直接对于每个质数，考虑有多少对会用到它就行了。

显然只要一个数的指数是奇数，另一个是偶数就行了。

设 $calc(n,p)$ 表示 $[1,n]$ 中有多少数 $p$ 的指数为奇数，显然有关系式：

$$calc(n,p)=\{\begin{array}{rlrl}& \lfloor n/p\rfloor -calc(\lfloor n/p\rfloor ,p)& & n\ge p^2\\ & \lfloor n/p\rfloor & & n<p^2\end{array}$$

而答案显然为

$$\sum _{p}p\cdot calc(n,p)\cdot (n-calc(n,p))$$

对于大于 $\sqrt{n}$ 的，min_25筛 处理一下区间素数之和即可，小于 $\sqrt{n}$ 的质数暴力枚举即可。

复杂度瓶颈在于处理区间素数之和，为 $O(n^{3/4}/\log n)$

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

using std::cerr;
using std::cout;

cs int mod=998244353;
inline int add(int a,int b){return a+b>=mod?a+b-mod:a+b;}
inline int dec(int a,int b){return a-b<0?a-b+mod:a-b;}
inline int mul(int a,int b){ll r=(ll)a*b;return r>=mod?r%mod:r;}
inline void Inc(int &a,int b){a+=b-mod;a+=a>>31&mod;}
inline void Dec(int &a,int b){a-=b;a+=a>>31&mod;}
inline void Mul(int &a,int b){a=mul(a,b);}
inline int po(int a,int b){int r=1;for(;b;b>>=1,Mul(a,a))if(b&1)Mul(r,a);return r;}
inline void ex_gcd(int a,int b,int &x,int &y){
	if(!b){x=1,y=0;return;}ex_gcd(b,a%b,y,x);y-=a/b*x;
}inline int inv(int a){int x,y;ex_gcd(mod,a,y,x);return x+(x>>31&mod);}
inline int MD(ll x){return x>=mod?x%mod:x;}

cs int N=1e6+7;

int p[N],pc;
int mu[N],sm[N];
bool mrk[N];

void linear_sieves(){
	sm[1]=1;mu[1]=1;
	for(int re i=2;i<N;++i){
		if(!mrk[i])p[++pc]=i,mu[i]=-1;
		sm[i]=sm[i-1]+!!mu[i];
		for(int re j=1;i*p[j]<N;++j){
			mrk[i*p[j]]=true;
			if(i%p[j]==0)break;
			mu[i*p[j]]=-mu[i];
		}
	}
}
int S(ll n){
	if(n<N)return sm[n];
	int ans=0,nn=sqrt(n);
	for(int re d=1;d<=nn;++d)if(mu[d])
		mu[d]==1?Inc(ans,MD(n/d/d)):Dec(ans,MD(n/d/d));
	return ans;
}

inline int Sum(ll x){
	x=MD(x);return MD(x*(x+1)>>1);
}
inline int C(ll x){
	x=MD(x);return MD(x*(x-1)>>1);
}

ll n;int nn;
int f1[N],f2[N];

void min_25_sieve(){
	nn=sqrt(n);
	for(int re i=1;i<=nn;++i)
		f1[i]=dec(Sum(i),1),f2[i]=dec(Sum(n/i),1);
	for(int re i=1;p[i]<=nn;++i){
		int p=::p[i],_f=f1[p-1];
		int w1=nn/p,w2=std::min((ll)nn,n/p/p);
		for(int re i=1;i<=w1;++i)
			Dec(f2[i],mul(p,dec(f2[i*p],_f)));
		for(int re i=w1+1;i<=w2;++i)
			Dec(f2[i],mul(p,dec(f1[n/i/p],_f)));
		for(int re i=nn;i>=(ll)p*p;--i)
			Dec(f1[i],mul(p,dec(f1[i/p],_f)));
	}
}

int calc(ll n,int p){
	if((ll)p*p>n)return MD(n/p);
	return MD(n/p-calc(n/p,p));
}

void Main(){
	linear_sieves();
	int T;scanf("%d",&T);
	while(T--){
		scanf("%lld",&n);
		int ans=0,las=0,now;
		for(ll l=1,r,u;(u=n/l);l=r+1)
			Inc(ans,mul(dec(now=S(r=n/u),las),C(sqrt(u)))),las=now;
		min_25_sieve();
		ll l=1,r,u;for(;(r=n/(n/l))<=nn;l=r+1);
		for(int re i=1;p[i]<=r;++i){
			int vl=calc(n,p[i]);
			Inc(ans,mul(p[i],mul(vl,MD(n-vl))));
		}for(l=r+1;(u=n/l);l=(r=n/u)+1){
			Inc(ans,mul(dec(f2[u],f2[u+1]),mul(MD(u),MD(n-u))));
		}cout<<ans<<"\n";
	}
}

inline void file(){
#ifdef zxyoi
	freopen("APSP.in","r",stdin);
#endif
}
signed main(){file();Main();return 0;}

---

如果这份代码要再快一点也很简单。

把所有下标的除法，拿出来，用 double 预处理倒数，计算乘法。

一般能快个2~4倍，不过没什么意义，懒得改了。