Codeforces 571D

做法很多，这里给一个比较优秀的仅使用并查集的做法。
不管是大学还是军队，它们的合并关系其实给定了两棵树，每次即是合并两个点。合并的时候，我们可以新建一个虚点，让原来两个点所在连通块的根连到虚点，这样标记不会错位。
这样如果仅有大学的操作，我们可以每次给某个连通块的根打一个加法标记，每次询问的即是某个点到连通块根路径的权值和，这个容易用带权并查集实现。现在加上了军队的操作，可以发现一个点我们仅需要知道一个点最后被覆盖的时间，就可以对大学操作离线后简单差分解决，而这个最后被覆盖的时间同样可以用类似的带权并查集求出。
时间复杂度$\mathcal{O}((n+m)\alpha (n))$。

#include <bits/stdc++.h>
#define FR first
#define SE second

using namespace std;

typedef long long ll;
typedef pair<int,int> pr;

namespace SETS1 {

int fa[1000005],maxn[1000005];

void init(int n) {
  for(int i=1;i<=n;i++) fa[i]=i;
}

int find_father(int x) {
  if (fa[x]==x) return x;
  int p=find_father(fa[x]);
  if (p!=fa[x]) {
  	maxn[x]=max(maxn[x],maxn[fa[x]]);
  	fa[x]=p;
  }
  return p;
}

void merge(int x,int y) {
  x=find_father(x);y=find_father(y);
  if (x==y) return;
  fa[x]=y;
}

int query(int x) {
  if (fa[x]==x) return maxn[x];
  else {
  	int p=find_father(x);
  	return max(maxn[x],maxn[p]);
  }
}

}

namespace SETS2 {

int fa[1000005];
ll sumv[1000005];

void init(int n) {
  for(int i=1;i<=n;i++) fa[i]=i;
}

int find_father(int x) {
  if (fa[x]==x) return x;
  int p=find_father(fa[x]);
  if (p!=fa[x]) {
  	sumv[x]+=sumv[fa[x]];
  	fa[x]=p;
  }
  return p;
}

void merge(int x,int y) {
  x=find_father(x);y=find_father(y);
  if (x==y) return;
  fa[x]=y;
}

ll query(int x) {
  if (fa[x]==x) return sumv[x];
  else {
  	int p=find_father(x);
    return sumv[x]+sumv[p];
  }
}

}

struct Query {
  int x,v,id;
  Query() {}
  Query(int a,int b,int c):x(a),v(b),id(c) {}
};

vector <Query> vt[500005];
ll ans[500005];

pr a[500005];
int size[500005];

int main() {
  int n,m;
  scanf("%d%d",&n,&m);
  SETS1::init(n+m);
  int sz=0;
  for(int i=1;i<=m;i++) {
  	char str[5];
  	scanf("%s",str);
  	if (str[0]=='U') {
  		int x,y;
  		scanf("%d%d",&x,&y);
  		a[i]=pr(x,y);
	  }
	else if (str[0]=='M') {
		int x,y;
		scanf("%d%d",&x,&y);
		SETS1::merge(x,n+i);
		SETS1::merge(y,n+i);
	}
	else if (str[0]=='A') {
		int x;
		scanf("%d",&x);
		a[i]=pr(x,0);
	}
	else if (str[0]=='Z') {
		int x;
		scanf("%d",&x);
		SETS1::maxn[SETS1::find_father(x)]=i;
	}
	else {
		int x;
		scanf("%d",&x);
		sz++;
		int t=SETS1::query(x);
		if (t) vt[t].push_back(Query(x,-1,sz));
		vt[i].push_back(Query(x,1,sz));
	}
  }
  for(int i=1;i<=n;i++) size[i]=1;
  SETS2::init(n+m);
  for(int i=1;i<=m;i++) {
  	if (a[i].FR&&a[i].SE) {
  		int x=a[i].FR,y=a[i].SE;
  		SETS2::merge(x,n+i);
  		SETS2::merge(y,n+i);
  	    size[x]+=size[y];
	  }
	else if (a[i].FR) {
		int x=a[i].FR;
		SETS2::sumv[SETS2::find_father(x)]+=size[x];
	}
	for(int j=0;j<vt[i].size();j++) {
	  Query t=vt[i][j];
	  ans[t.id]+=SETS2::query(t.x)*t.v;
    }
  }
  for(int i=1;i<=sz;i++) printf("%lld\n",ans[i]);
  return 0;
}