bzoj3237 [Ahoi2013]連通圖 線段樹分治+並查集按秩合併
阿新 • • 發佈:2018-12-30
Description
給定n個點m條邊的無向圖,k次詢問,每次刪除s條邊並詢問此時圖的連通性,詢問互相獨立。
n<=1e5,m<=2e5,k<=1e5,s<=4
Solution
傳說中的線段樹分治
刪除和插入同時存在的話非常麻煩,因此考慮一種處理詢問的順序使得只剩插入操作
我們把一條邊存在的詢問區間扔進線段樹,然後遍歷整棵線段樹。遍歷的時候就插入區間內的邊,同時壓進一個棧裡。回溯的時候按照插入順序的逆序刪除並彈棧。這樣在葉子處判斷就可以了。寫起來就像另一個版本的cdq
判斷圖連通可以並查集維護集合size,要求資瓷可撤銷可以可持久化也可以按秩合併,高興就好
Code
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
#include <stack>
#define rep(i,st,ed) for (register int i=st;i<=ed;++i)
#define max(x,y) ((x)>(y)?(x):(y))
#define min(x,y) ((x)<(y)?(x):(y))
const int N=200005;
struct edge { int x,y;} ;
int fa[N],size[N],d[N],pre[N],h[N],top,n;
std:: vector <int> vec[N<<2];
edge e[N],stack[N];
int read() {
int x=0,v=1; char ch=getchar();
for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
return x* v;
}
int find(int x) {
return (!fa[x])?(x):(find(fa[x]));
}
void ins(int now,int tl,int tr,int l,int r,int x) {
if (r<l) return ;
if (tl>=l&&tr<=r) {
vec[now].push_back(x);
return ;
}
int mid=(tl+tr)>>1;
ins(now<<1,tl,mid,l,min(r,mid),x);
ins(now<<1|1,mid+1,tr,max(mid+1,l),r,x);
}
void solve(int now,int l,int r) {
int wjp=top;
for (int i=0,_=vec[now].size();i<_;++i) {
edge ed=e[vec[now][i]];
int x=find(ed.x),y=find(ed.y);
if (x==y) continue;
if (h[x]<h[y]) fa[x]=y,size[y]+=size[x],stack[++top]=(edge) {x,0};
else if (h[x]>h[y]) fa[y]=x,size[x]+=size[y],stack[++top]=(edge) {y,0};
else if (h[x]==h[y]) {
fa[y]=x,size[x]+=size[y],h[x]++,stack[++top]=(edge) {y,1};
}
}
if (l==r) {
puts((size[find(1)]==n)?"Connected":"Disconnected");
} else {
int mid=(l+r)>>1;
solve(now<<1,l,mid);
solve(now<<1|1,mid+1,r);
}
for (;top>wjp;) {
edge lxf=stack[top--];
int ff=fa[lxf.x],x=lxf.x; fa[x]=0;
size[ff]-=size[x];
if (lxf.y) h[ff]--;
}
}
int main(void) {
freopen("data.in","r",stdin);
freopen("myp.out","w",stdout);
n=read(); int m=read();
rep(i,1,n) size[i]=1;
rep(i,1,m) {
e[i].x=read(),e[i].y=read();
pre[i]=1;
}
int T=read();
rep(i,1,T) {
for (int k=read();k--;) {
int x=read();
if (pre[x]<i) ins(1,1,T,pre[x],i-1,x);
pre[x]=i+1;
}
}
rep(i,1,m) if (pre[i]<=T) ins(1,1,T,pre[i],T,i);
solve(1,1,T);
return 0;
}