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