「Codeforces 643D」Bearish Fanpages

传送门

---

problem

小 A 所在的地区有 $n$ 个点，标号为 $1\sim n$。每个节点都连出去恰好一条有向边，设 $i$ 连出去的点是 $A_i$。保证 $A_i\ne i$ 且 $A_{A_i}\ne i$。

每个点上有一些糖果，第 $i$ 个节点上的糖果数量为 $B_i$，小 A 定义一个节点的稠密度为 $C_i$，$C_i$ 求法如下：

假设和 $i$ 距离不超过 $1$ 的点有 $D_i$ 个（包括 $i$ 连出去的点、连向 $i$ 的点以及 $i$ 自己），分别是 $P_1,P_2,...P_{D_i}$。

设 $E_i=\lfloor \frac{B_i}{D_i}\rfloor$，那么 $C_i=B_i-D_i\times E_i+{\sum }_{j=1}^{D_i}{E}_{P_j}$。

现在小 A 想让你实现一个糖果稠密度分析仪，要支持三种操作：

• $1$ $i$ $j$：表示把 $i$ 的出边改为 $j$，即令 $Ai=j$，保证 $j\ne i$ 且 $A_j\ne i$。
• $2$ $i$：表示询问 $i$ 点的稠密度，即你需要输出 $C_i$。
• $3$：询问所有节点中，$C_i$ 的最小值和最大值。

数据范围：$3\le n\le 10^5$，$1\le q\le 10^5$，$1\le B_i\le 10^{12}$，$1\le A_i\le n$。

---

solution

感觉这道题想通了就不算很难了，主要是代码细节问题。

把 $C_i$ 中 $A_i$ 的贡献与其他点的贡献分开，即我们先不考虑 $A_i$，最后加上$E_{A_i}$。

记集合 s o n i = { C j ∣ A j = i } son_i=\{C_j \mid A_j=i\} soni​={Cj​∣Aj​=i}（注意这里的 $C_j$ 也是不考虑 $E_{A_j}$ 的）。对每个点都维护一个这样的集合，然后把 E i + m a x { s o n i } E_i+max\{son_i\} Ei​+max{soni​} 和 E i + m i n { s o n i } E_i+min\{son_i\} Ei​+min{soni​} 丢到全局的集合，最后全局集合的最大最小值就是 $3$ 的答案。

对于 $2$，答案也就是 $C_x+E_{A_x}$。

那么对于 $1$ 的修改，考虑会有这些东西的更改：

• $i$ 点的 $A$ 值。
• 原 $A_i$ 点的 $C,D,E$ 值和 $son$ 集合。
• $A_{A_i}$ 点的 $C$ 值和 $son$ 集合。
• $A_{A_{A_i}}$ 点的 $son$ 集合。
• $j$ 点的 $C,D,E$ 值和 $son$ 集合。
• $A_j$ 的 $C$ 值和 $son$ 集合。
• $A_{A_j}$ 的 $son$ 集合。

这些其实在草稿纸上好好推一推就可以得到，具体怎么改根据公式也很容易推出。

时间复杂度 $O(n\log n)$。

---

code

#include<bits/stdc++.h>
#define N 100005
#define ll long long
using namespace std;
int n,q;
ll A[N],B[N],C[N],D[N],E[N];
multiset<ll>son[N],Ans;
void Insert(int x){
	if(!son[x].empty()){
		Ans.insert(E[x]+*son[x].begin());
		Ans.insert(E[x]+*son[x].rbegin());
	}
}
void Delete(int x){
	if(!son[x].empty()){
		Ans.erase(Ans.find(E[x]+*son[x].begin()));
		Ans.erase(Ans.find(E[x]+*son[x].rbegin()));
	}
}
void ins(int fa,int x)  {son[fa].insert(C[x]);}
void del(int fa,int x)  {son[fa].erase(son[fa].find(C[x]));}
void Modify(int i,int j){
	int x=A[i],y=A[x],z=A[y];
	Delete(x),Delete(y),Delete(z);del(z,y),del(y,x),del(x,i);
	C[y]-=E[x],C[x]-=E[i]-E[x]*(D[x]-1);
	--D[x],E[x]=B[x]/D[x];
	C[y]+=E[x],C[x]-=E[x]*(D[x]-1);
	ins(y,x),ins(z,y);Insert(x),Insert(y),Insert(z);
	
	A[i]=j,x=A[j],y=A[x];
	Delete(j),Delete(x),Delete(y);del(y,x),del(x,j);
	C[x]-=E[j],C[j]+=E[j]*(D[j]-1);
	++D[j],E[j]=B[j]/D[j];
	C[x]+=E[j],C[j]-=E[j]*(D[j]-1)-E[i];
	ins(j,i),ins(x,j),ins(y,x);Insert(j),Insert(x),Insert(y);
}
int main(){
	scanf("%d%d",&n,&q);
	for(int i=1;i<=n;++i)  scanf("%lld",&B[i]),D[i]=2;
	for(int i=1;i<=n;++i)  scanf("%lld",&A[i]),D[A[i]]++;
	for(int i=1;i<=n;++i)  E[i]=B[i]/D[i];
	for(int i=1;i<=n;++i)  C[i]+=B[i]-E[i]*(D[i]-1),C[A[i]]+=E[i];
	for(int i=1;i<=n;++i)  ins(A[i],i);
	for(int i=1;i<=n;++i)  Insert(i);
	int op,x,y;
	while(q--){
		scanf("%d",&op);
		if(op==1)  scanf("%d%d",&x,&y),Modify(x,y);
		else  if(op==2)  scanf("%d",&x),printf("%lld\n",C[x]+E[A[x]]);
		else  printf("%lld %lld\n",*Ans.begin(),*Ans.rbegin());
	}
	return 0;
}