LOJ.121.[离线可过]动态图连通性(线段树分治 按秩合并)

题目链接

以时间为下标建线段树。线段树每个节点开个vector。
对每条边在其出现时间内加入线段树，即，把这条边按时间放在线段树的对应区间上，会影响\(O(\log n)\)个节点。
询问就放在线段树的对应叶子节点上。

然后对整棵树DFS，当进入一个节点时，将这个点代表的这段区间中出现的边全部加到图里，即合并起来，之后在离开这个点时撤销。
可以用不路径压缩、按秩合并的并查集维护连通性。这样就可以撤销了。
合并时用栈记录合并前状态，合并前的父节点用\(x\)或是\(fa[x]\)都行，因为合并的时候是合并集合根节点，用\(x\)还是\(fa[x]\)都一样。。
如果秩改变也要加入栈。

启发式合并按深度还是按size无所谓。

//1548ms    48256K
#include <map>
#include <cstdio>
#include <cctype>
#include <vector>
#include <algorithm>
#define mp std::make_pair
#define pr std::pair<int,int>
#define Vec std::vector<int>
//#define gc() getchar()
#define MAXIN 500000
#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
const int N=5005,M=5e5+5;

int n,m,fa[N],dep[N],top;
std::map<pr,int> las;
char IN[MAXIN],*SS=IN,*TT=IN;
struct Node
{
    int x,dep;
}sk[M<<1];//M就够吧。。
struct Operation
{
    int id,u,v;
}opt[M];
struct Segment_Tree
{
    #define lson rt<<1
    #define rson rt<<1|1
    std::vector<int> vec[M<<2];
    void Update(int l,int r,int rt,int L,int R,int id)
    {
        if(L<=l && r<=R) return (void)vec[rt].push_back(id);
        int m=l+r>>1;
        if(L<=m) Update(l,m,rt<<1,L,R,id);
        if(m<R) Update(m+1,r,rt<<1|1,L,R,id);
    }
}T;

inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-'0',c=gc());
    return now;
}
int Get_fa(int x){
    return x==fa[x]?x:Get_fa(fa[x]);
}
inline void Merge(int x,int y)
{
    if((x=Get_fa(x))==(y=Get_fa(y))) return;
    if(dep[y]>dep[x]) std::swap(x,y);//y->x
    fa[y]=x, sk[++top]=(Node){y,dep[y]};
    if(dep[x]==dep[y]) sk[++top]=(Node){x,dep[x]++};
}
inline void Delete(int bef)
{
    while(top>bef)
    {
        int x=sk[top].x;
        fa[x]=x, dep[x]=sk[top--].dep;
    }
}
void DFS(int l,int r,int rt)
{
    int now=top;
    const Vec &v=T.vec[rt];
    for(int i=0,lim=v.size(); i<lim; ++i) Merge(opt[v[i]].u,opt[v[i]].v);
    if(l==r){
        if(opt[l].id==2)//询问直接在叶节点上 
            putchar(Get_fa(opt[l].u)==Get_fa(opt[l].v)?'Y':'N'),putchar('\n');
    }
    else DFS(l,l+r>>1,lson), DFS((l+r>>1)+1,r,rson);
    Delete(now);
}

int main()
{
    n=read(), m=read();
    for(int i=1; i<=n; ++i) fa[i]=i;
    std::map<pr,int>::iterator it;
    for(int id,x,y,i=1; i<=m; ++i)
    {
        id=read(), x=read(), y=read();
        if(x>y) std::swap(x,y);
        if(!id) las[mp(x,y)]=i;
        else if(id==1) it=las.find(mp(x,y)), T.Update(1,m,1,it->second,i,i), it->second=0;//包含不包含i无所谓了，i处已不是询问 
        opt[i]=(Operation){id,x,y};
    }
    for(it=las.begin(); it!=las.end(); ++it)
        if(it->second) T.Update(1,m,1,it->second,m,it->second);
    DFS(1,m,1);

    return 0;
}

转载于:https://www.cnblogs.com/SovietPower/p/9401959.html