1. 程式人生 > >bzoj3237 [Ahoi2013]連通圖 線段樹分治+並查集按秩合併

bzoj3237 [Ahoi2013]連通圖 線段樹分治+並查集按秩合併

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; }