「Codeforces 809E」Surprise me!

传送门

---

problem

给出一个 $n$ 个点的树，每个点有一个点权 $a_i$，且 $a_i$ 构成 $1\sim n$ 的一个排列，求：

$$\frac{1}{n(n-1)}\sum _{i=1}^n\sum _{j=1}^n\phi (a_i{a}_j)\times \mathrm{d}\mathrm{i}\mathrm{s}(i,j)$$

其中 $\phi$ 是欧拉函数，$dis(i,j)$ 表示 $i,j$ 两点在树上的距离。答案对 $10^9+7$ 取模。

数据范围：$2\le n\le 2\times 10^5$。

---

solution

这也是道神题啊 orz。

首先，欧拉函数有一个性质：

$$\phi (ij)=\phi (i)\phi (j)\frac{\gcd (i,j)}{\phi (\gcd (i,j))}$$

这个根据 $\phi (n)=n(1-\frac{1}{p_1})(1-\frac{1}{p_2})\cdots (1-\frac{1}{p_k})$ 即可推出。

先把 $\frac{1}{n(n-1)}$ 抛开，考虑求：

$$ans=\sum _{i=1}^n\sum _{j=1}^n\phi (a_i)\phi (a_j)\frac{\gcd (a_i,a_j)}{\phi (\gcd (a_i,a_j))}\times \mathrm{d}\mathrm{i}\mathrm{s}(i,j)$$

枚举 $\gcd$，得到：

$$ans=\sum _{k=1}^n\frac{k}{\phi (k)}\sum _{i=1}^n\sum _{j=1}^n[\gcd (a_i,a_j)=k]\phi (a_i)\phi (a_j)\times \mathrm{d}\mathrm{i}\mathrm{s}(i,j)$$

我们做一个小转化，设值 $i$ 对应的点为 $p_i$，则：

$$\begin{array}{rl}ans& =\sum _{k=1}^n\frac{k}{\phi (k)}\sum _{i=1}^n\sum _{j=1}^n[\gcd (i,j)=k]\phi (i)\phi (j)\times \mathrm{d}\mathrm{i}\mathrm{s}(p_i,p_j)\\ & =\sum _{k=1}^n\frac{k}{\phi (k)}\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\sum _{j=1}^{\lfloor \frac{n}{k}\rfloor }[\gcd (i,j)=1]\phi (ik)\phi (jk)\times \mathrm{d}\mathrm{i}\mathrm{s}(p_{ik},p_{jk})\end{array}$$

然后就可以莫反了：

$$\begin{array}{rl}ans& =\sum _{k=1}^n\frac{k}{\phi (k)}\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\sum _{j=1}^{\lfloor \frac{n}{k}\rfloor }\sum _{d|\gcd (i,j)}\mu (d)\phi (ik)\phi (jk)\times \mathrm{d}\mathrm{i}\mathrm{s}(p_{ik},p_{jk})\\ & =\sum _{k=1}^n\frac{k}{\phi (k)}\sum _{d=1}^{\lfloor \frac{n}{k}\rfloor }\mu (d)\sum _{i=1}^{\lfloor \frac{n}{kd}\rfloor }\sum _{j=1}^{\lfloor \frac{n}{kd}\rfloor }\phi (ikd)\phi (jkd)\times \mathrm{d}\mathrm{i}\mathrm{s}(p_{ikd},p_{jkd})\end{array}$$

令 $T=kd$，我们枚举 $T$，得：

$$ans=\sum _{T=1}^n\sum _{k|T}\frac{\mu (\frac{T}{k})k}{\phi (k)}\sum _{i=1}^{\lfloor \frac{n}{T}\rfloor }\sum _{j=1}^{\lfloor \frac{n}{T}\rfloor }\phi (iT)\phi (jT)\times \mathrm{d}\mathrm{i}\mathrm{s}(p_{iT},p_{jT})$$

设 $f(n)={\sum }_{k|n}\frac{\mu (\frac{n}{k})k}{\phi (k)}$，这个可以枚举倍数在 $O(n\log n)$ 的时间内算出来。

考虑怎么计算后面的那个式子，由于每次涉及的点只有 $O(\lfloor \frac{n}{T}\rfloor )$ 个，就想到可以在虚树上 $dp$。

具体就是，把虚树建出来之后，考虑一下每条边的贡献就可以了。

PS：注意最后别忘了除以 $n(n-1)$。

PPS：其实感觉这道题莫反挺套路的，关键是要想到那个转化和在虚树上 $dp$。

时间复杂度 $O(n{\log }^{2}n)$。

---

code

#include<bits/stdc++.h>
#define N 200005
#define P 1000000007
using namespace std;
int n,tot;
int a[N],pos[N],dfn[N],dep[N],fa[N][21];
struct edge{
	int t,first[N],v[N<<1],nxt[N<<1];
	void add(int x,int y){
		nxt[++t]=first[x],first[x]=t,v[t]=y;
	}
}T,VT;
int add(int x,int y)  {return x+y>=P?x+y-P:x+y;}
int dec(int x,int y)  {return x-y< 0?x-y+P:x-y;}
int mul(int x,int y)  {return 1ll*x*y%P;}
int power(int a,int b){
	int ans=1;
	for(;b;b>>=1,a=mul(a,a))  if(b&1)  ans=mul(ans,a);
	return ans;
}
int inv(int x)  {return power(x,P-2);}
bool comp(const int &p,const int &q)  {return dfn[p]<dfn[q];}
int sum,prime[N],mark[N],mu[N],phi[N],f[N];
void linear_sieves(){
	phi[1]=mu[1]=1;
	for(int i=2;i<N;++i){
		if(!mark[i])  prime[++sum]=i,mu[i]=P-1,phi[i]=i-1;
		for(int j=1;j<=sum&&i*prime[j]<N;++j){
			mark[i*prime[j]]=1;
			if(i%prime[j])  mu[i*prime[j]]=P-mu[i],phi[i*prime[j]]=phi[i]*(prime[j]-1);
			else  {mu[i*prime[j]]=0,phi[i*prime[j]]=phi[i]*prime[j];break;}
		}
	}
	for(int i=1;i<N;++i){
		int Inv=inv(phi[i]);
		for(int j=1;i*j<N;++j)
			f[i*j]=add(f[i*j],mul(mul(i,mu[j]),Inv));
	}
}
void dfs(int x){
	dfn[x]=++tot;
	for(int i=1;i<=20;++i)
		fa[x][i]=fa[fa[x][i-1]][i-1];
	for(int i=T.first[x];i;i=T.nxt[i]){
		int to=T.v[i];
		if(to==fa[x][0])  continue;
		fa[to][0]=x,dep[to]=dep[x]+1;
		dfs(to);
	}
}
int LCA(int x,int y){
	if(dep[x]<dep[y])  swap(x,y);
	for(int i=20;~i;--i)  if(dep[fa[x][i]]>=dep[y])  x=fa[x][i];
	if(x==y)  return x;
	for(int i=20;~i;--i)  if(fa[x][i]!=fa[y][i])  x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
namespace Virtual_Tree{
	int sum,res,cnt,top;
	int p[N],S[N],stk[N];
	void Insert(int x){
		if(top==1)  {stk[++top]=x;return;}
		int lca=LCA(x,stk[top]);
		if(lca==stk[top])  {stk[++top]=x;return;}
		while(top>1&&dfn[lca]<=dfn[stk[top-1]])
			VT.add(stk[top-1],stk[top]),top--;
		if(lca!=stk[top])  VT.add(lca,stk[top]),stk[top]=lca;
		stk[++top]=x;
	}
	int dis(int x,int y)  {return dep[x]+dep[y]-2*dep[LCA(x,y)];}
	void dp(int x){
		for(int i=VT.first[x];i;i=VT.nxt[i]){
			int to=VT.v[i],len=dis(x,to);
			dp(to);
			res=add(res,mul(len,mul(dec(sum,S[to]),S[to])));
			S[x]=add(S[x],S[to]);
		}
	}
	void Clear(int x){
		for(int i=VT.first[x];i;i=VT.nxt[i])  Clear(VT.v[i]);
		S[x]=VT.first[x]=0;
	}
	int calc(int x){
		VT.t=cnt=sum=res=0;
		for(int i=1;i*x<=n;++i)  p[++cnt]=pos[i*x],S[p[cnt]]=phi[i*x],sum=add(sum,phi[i*x]);
		sort(p+1,p+cnt+1,comp),stk[top=1]=1;
		for(int i=1;i<=cnt;++i)  if(p[i]!=1)  Insert(p[i]);
		while(--top)  VT.add(stk[top],stk[top+1]);
		dp(1),Clear(1);
		res=mul(add(res,res),f[x]);
		return res;
	}
}
using namespace Virtual_Tree;
int main(){
	scanf("%d",&n),linear_sieves();
	for(int i=1;i<=n;++i)  scanf("%d",&a[i]),pos[a[i]]=i;
	for(int i=1,x,y;i<n;++i){
		scanf("%d%d",&x,&y);
		T.add(x,y),T.add(y,x);
	}
	dep[1]=1,dfs(1);
	int ans=0;
	for(int i=1;i<=n;++i)  ans=add(ans,calc(i));
	printf("%d\n",mul(ans,inv(mul(n,n-1))));
	return 0;
}