NowCoder 14648 序列(莫比乌斯反演)

NowCoder 14648 序列

Statement

两个长度为$n$的序列$a,b$, 求满足$\gcd (x,y)=1$且$a_{b_x}=b_{a_y}$的的对$(x,y)$数量

Solution

$Ans=\sum _{1\le x\le n且1\le y\le n}[a_{b_x}=b_{a_y}]\cdot [\gcd (x,y)=1]$.

• $f(d)=\sum _{1\le x\le n且1\le y\le n}[a_{b_x}=b_{a_y}][\gcd (x,y)=d]$.

• $g(d)={\sum }_{d|d^{\prime }}f(d^{\prime })=\sum _{d|x且d|y}[a_{b_x}=b_{a_y}]$.

  • $g(d)$的组合含义是, 我不妨枚举$d^{\prime }$满足$d|d^{\prime }$那么$d|x$且$d|y$时当且仅当$d=\gcd (x,y)$时才满足条件$[\gcd (x,y)=d]=true$. 否则为$false$. 那么显然一共$[a_{b_x}=b_{a_y}]$只会在$d^{\prime }=\gcd (x,y)$统计一次.
  • 因此我们得到一个等价式$g(d)=\sum _{d|x且d|y}[a_{b_x}=b_{a_y}]$.

---

我们再考虑莫比乌斯反演的本质: 莫比乌斯反演本质上运用了容斥原理, 当我们用容斥原理蓝理解莫比乌斯反演的时候:

以$\phi (n)={\sum }_{d|n}\mu (d)\frac{n}{d}$为例: 考虑不超过$n$的正整数$k$, 根据欧拉函数定义, 我们要计算出有多少个$k$与$n$互质.

考虑质数集合 P = { p ∣ p 是 n 的 质 因 子 } P=\{p|p是n的质因子\} P={p∣p是n的质因子}, 当$\gcd (n,k)>1$是, 显然$k$会是$P$中某些质数的倍数. 这些数都要被去掉. 但很容易发现这样做是很容易重复计数的.

继续考虑两个不同质数$p_1,p_2(p_1,p_2\in P)$, 若$p_1{p}_2|k$, 这样的$k$在之前的操作中扣除了$C_2^1$次, 因为$\frac{n}{p_1{p}_2}$个$k$被多扣除了一次, 所以答案需要加上$\frac{n}{p_1{p}_2}$. 继续考虑三个不同质数$p_1,p_2,p_3(p_1,p_2,p_3\in P)$的情况, 若$p_1{p}_2{p}_3|k$, 则这样的$k$在之前操作已经扣除了$C_3^1-C_3^2$次, 因此答案需要再次减去$\frac{n}{p_1{p}_2{p}_3}$.

以此类推, 考虑$t$个不同质数$p_1,p_2\dots ,p_t(p_1,p_2,\dots ,p_t\in P)$, 若$p_1{p}_2{p}_3\dots p_t|k$, 这样的$k$在之前操作中被扣除了$C_t^1-C_t^2+C_t^3+\cdots +(-1{)}^t{C}_t^{t-1}=\{\begin{array}{ll}2& t是偶数\\ 0& t是奇数\end{array}$. 然后我们希望$k$被减去$1$次. 所以呢可以得到$t$是偶数时应该加上$\frac{n}{p_1{p}_2\dots p_n}$若$t$是奇数时应该减去$\frac{n}{p_1{p}_2\dots p_n}$. 那么令$d=p_1{p}_2{p}_3\dots p_t$那么对于欧拉函数而言, 答案就要加上$\mu (d)\frac{n}{d}$.

---

对于这道题 我们靠考虑${\sum }_{d=1}^{n}g(d)$的含义, 显然对于对于每一对数$x,y$, 将其$\gcd (x,y)=p_1^{s_1}{p}_2^{s_2}\dots p_r^{s_r}$. 我们考虑对每个质因子进行容斥, 从而达到满足$\gcd (x,y)=1$的条件, 类似的我们可以发现如果$p_1|\gcd (x,y)$则我们需要将其减去$1$次, 若不同的质因子$p_1{p}_2|\gcd (x,y)$则扣除了$C_2^1$次我们需要重新加上和上述容斥过程一致$\dots$我们就可以得到对于一个数$d=p_1{p}_2\dots p_r$它对于答案的贡献就位$\mu (d)=(-1{)}^r$. 因为上述容斥式子, 我们均考虑只要包含该质因子就好被筛去, 所以显然平方、立方因子也会被筛去, 在$s_i=1$时被恰好筛去了一次. 故不会重复扣除, 就没必要重复容斥了.

所以我们得到了这个式子: $Ans=f(1)=\sum _{d=1}^n\mu (d)g(d)$. 时间复杂度$O({\sum }_{i=1}^n\frac{n}{i})=O(n\log n)$.

Code

# define Fast_IO std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
# include "unordered_map"
# include "algorithm"
# include "iostream"
# include "cstdlib"
# include "cstring"
# include "cstdio"
# include "vector"
# include "bitset"
# include "queue"
# include "cmath"
# include "ctime"
# include "map"
# include "set"
# define ll long long
# define ld long double
# define rep1(i,a,b) for(ll i=(a);i<=(b);i++)
# define rep2(i,a,b) for(ll i=(b);i>=(a);i--)
# define pii pair<int,int>
# define pll pair<ll,ll>
# define ph push_back
# define pb pop_back
# define eb emplace_back
# define vi vector<int>
# define vll vector<ll>
# define vpi vector<pii >
# define vpll vector<pll >
# define ri(x) scanf("%d",&x)
# define rf(x) scanf("%f",&x)
# define rl(x) scanf("%lld",&x)
# define rd(x) scanf("%lf",&x)
# define rs(s) scanf("%s",s+1)
# define wi(x) printf("%d",x)
# define wl(x) printf("%lld",x)
# define ws(s) printf("%s",s+1)
# define resize(v,x) v.resize(x)
# define all(v) v.begin(),v.end()
# define reverse(v) reverse(all(v))
# define fi first
# define se second
# define lowbit(x) ((x)&(-(x)))
# define repauto(Name,v) for(auto Name:v)
# define Endl "\n"
using namespace std;
template<class I> inline I GCD(I A,I B){return B?GCD(B,A%B):A;}
template<class I> inline I LCM(I A,I B){return A/GCD(A,B)*B;}
template<class I> I Sqrt(I N){
	I sqrtN=sqrt(N)-1;
	while(sqrtN+1<=N/(sqrtN+1))sqrtN++;
	return sqrtN;
}
template<class I> I Pow(I X,I Y,__int128 Mod1=998244353){
	static __int128 Ans; Ans=1;
	for(;Y;Y>>=1,X=(__int128)X*X%Mod1) if(Y&1) Ans=Ans*X%Mod1;
	return Ans;
}

namespace Mu_Class{
	int Prime_Cnt;
	int *Prime=new int;
	int *Mu=new int;
	int *Visit=new int;
	int *Num=new int;
	inline void Init(int N){
		static int i,j,Now;
		delete Visit;	Visit=new int[N+1]();
		delete Prime;	Prime=new int[N+1]();
		delete Mu;		Mu=new int[N+1]();
		delete Num;		Num=new int[N+1]();
		for(Mu[1]=1,i=2;i<=N;i++){
			if(!Visit[i]){
				Visit[i]=i,Mu[i]=-1;
				Prime[++Prime_Cnt]=i;
			}for(j=1;j<=Prime_Cnt && Visit[i]>=Prime[j] && Prime[j]<=N/i;j++){
				Mu[i*Prime[j]]=Visit[i]==Prime[j]?0:Mu[i]*(-1);
				Visit[i*Prime[j]]=Prime[j];
			}
		}return;
	}
}using namespace Mu_Class;

const int maxm=1e5+10;

int N;
int A[maxm],B[maxm];
long long Ans;
unordered_map<int,int> Visit1;

long long Solve(int Now){
	static int i;
	static long long Res;
	Visit1.clear();
	for(i=Now;i<=N;i+=Now) Visit1[A[B[i]]]++;
	for(i=Now,Res=0;i<=N;i+=Now) Res+=Visit1[B[A[i]]];
	return Res;
}

int main(){
# ifdef LH_Frank
    freopen("1.in","r",stdin);
	freopen("1.out","w",stdout);
# endif
	ri(N);
	rep1(i,1,N) ri(A[i]);
	rep1(i,1,N) ri(B[i]);
	Init(N);
	rep1(i,1,N){
		if(Mu[i]==0) continue;
		Ans+=Mu[i]*Solve(i);
	}wl(Ans);
	return 0;
}

Link

[1] NowCoder 序列