1. 程式人生 > >[BZOJ1018][SHOI2008]堵塞的交通traffic

[BZOJ1018][SHOI2008]堵塞的交通traffic

記得 lin clas 做出 define wap 錯誤筆記 err end

題面

介紹一種比較慢的但是好想的做法。

網上漫天的線段樹維護聯通性,然後想起來費很大周折也很麻煩。我的做法也是要用線段樹的,不過用法完全不同。

這個東西叫做時間分治線段樹。

首先我們建一個\(1..m+1\)的線段樹。

很好做出每條邊的存在時間的區間是吧,所以我們這段時間存入線段樹中。(最後都沒有消失的視為\(m+1\)時間消失)記錄下每個節點的對應區間的所有邊。

然後從上往下掃整個線段樹,將該段區間的邊用並查集維護連通性。遇到詢問就查一下。

回去的時候還有撤掉之前連的邊,所以並查集給用只有按秩合並,不帶按秩合並。這樣並查集一次的操作時間復雜度\(O(\log n)\)

總時間復雜度\(O(n\log ^2 n)\)

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<map>
#define lc o<<1
#define rc o<<1|1
#define REP(i,a,n) for(register int i(a);i<=(n);++i)
#define PER(i,a,n) for(register int i(a);i>=(n);--i)
#define FEC(i,x,y) for(register int i=head[x],y=g[i].to;i;i=g[i].ne,y=g[i].to)
#define dbg(...) fprintf(stderr,__VA_ARGS__)
const int SZ=(1<<21)+1;char ibuf[SZ],*iS,*iT,obuf[SZ+128],*oS=obuf,*oT=obuf+SZ-1;
#ifdef ONLINE_JUDGE
#define gc() (iS==iT?(iT=(iS=ibuf)+fread(ibuf,1,SZ,stdin),(iS==iT?EOF:*iS++)):*iS++)
#else
#define gc() getchar()
#endif
template<typename I>inline void read(I&x){char c=gc();int f=0;for(;c<'0'||c>'9';c=gc())c=='-'?f=1:0;for(x=0;c>='0'&&c<='9';c=gc())x=(x<<1)+(x<<3)+(c&15);f?x=-x:0;}
template<typename I>inline void read(I*s){char c=gc();for(;c<=' ';c=gc());for(;c>' ';c=gc())*s++=c;*s='\0';}
inline void flush(){fwrite(obuf,1,oS-obuf,stdout);oS=obuf;}
#define printf(...) (oS>oT&&(flush(),1),oS+=sprintf(oS,__VA_ARGS__))
#define putchar(x) (oS>oT&&(flush(),1),*oS++=(x))
template<typename A,typename B>inline char SMAX(A&a,const B&b){return a<b?a=b,1:0;}
template<typename A,typename B>inline char SMIN(A&a,const B&b){return a>b?a=b,1:0;}
typedef long long ll;typedef unsigned long long ull;typedef std::pair<int,int>pii;

const int N=200000+7;
int n,m,x1,y1,x2,y2,id[3][N],ans[N],ext[N];char opt[7];

int fa[N],siz[N];
inline int Find(int x){return fa[x]==x?x:Find(fa[x]);}
inline void Union(int x,int y){x=Find(x),y=Find(y);if(siz[x]<siz[y])std::swap(x,y);SMAX(siz[x],siz[y]+1);fa[y]=x;}

struct QUE{int opt,x,y,prex,prey,sizx,sizy;}q[N<<1];std::vector<QUE>t[N<<1];
inline void Insert(int o,int L,int R,int l,int r,QUE k){
    if(l<=L&&R<=r)return t[o].push_back(k);
    int M=(L+R)>>1;if(l<=M)Insert(lc,L,M,l,r,k);if(r>M)Insert(rc,M+1,R,l,r,k);
}
inline void Insert(int o,int L,int R,int x,QUE k){
    if(L==R)return t[o].push_back(k);
    int M=(L+R)>>1;x<=M?Insert(lc,L,M,x,k):Insert(rc,M+1,R,x,k);
}
inline void Solve(int o,int L,int R){
    int len=t[o].size();REP(i,0,len-1)if(t[o][i].opt){const int&x=Find(t[o][i].x),y=Find(t[o][i].y);if(x==y)continue;t[o][i].prex=fa[x],t[o][i].prey=fa[y];t[o][i].sizx=siz[x],t[o][i].sizy=siz[y];Union(x,y);}
    if(L==R){REP(i,0,len-1)if(!t[o][i].opt)ans[t[o][i].prex]=Find(t[o][i].x)==Find(t[o][i].y);}
    else{int M=(L+R)>>1;Solve(lc,L,M);Solve(rc,M+1,R);}
    PER(i,len-1,0)if(t[o][i].opt){const int&x=t[o][i].prex,&y=t[o][i].prey;fa[x]=x,fa[y]=y,siz[x]=t[o][i].sizx,siz[y]=t[o][i].sizy;}//錯誤筆記:和第44行一樣,這裏的prex啥的應該存下來其find的信息不是自己的信息
}

std::map<pii,int>mp;
int main(){
    read(n);REP(i,1,n)id[1][i]=i,id[2][i]=n+i,fa[i]=i,fa[n+i]=n+i,siz[i]=siz[n+i]=1;n<<1;//錯誤筆記:並查集要記得初始化
    REP(i,1,N){
        read(opt);if(*opt=='E'){m=i-1;break;}
        read(x1),read(y1),read(x2),read(y2);q[i]=QUE{(int)*opt,id[x1][y1],id[x2][y2]};
    }
    REP(i,1,m){
        int opt=q[i].opt,id1=q[i].x,id2=q[i].y;
        if(opt=='O')mp[pii(id1,id2)]=i;
        else if(opt=='C'){
            pii e(id1,id2);
            Insert(1,1,m+1,mp[e],i,QUE{1,id1,id2});mp.erase(e);
        }else if(opt=='A')Insert(1,1,m+1,i,QUE{0,id1,id2,++ans[0]});
    }
    std::map<pii,int>::iterator p=mp.begin();
    while(p!=mp.end()){Insert(1,1,m+1,p->second,m+1,QUE{1,p->first.first,p->first.second});++p;}
    Solve(1,1,m+1);
    REP(i,1,ans[0])putchar(ans[i]?'Y':'N'),putchar('\n');
    return flush(),0;
}

[BZOJ1018][SHOI2008]堵塞的交通traffic