BZOJ-3674:可持久化並查集(可持久化數據結構)

 

題目大意:

中文題目,不多解釋。

n個集合 m個操作
操作:
1 a b 合併a,b所在集合
2 k 回到第k次操作之後的狀態(查詢算作操作)
3 a b 詢問a,b是否屬於同一集合,是則輸出1否則輸出0
請注意本題採用強制在線,所給的a,b,k均經過加密,加密方法爲x = x xor lastans,lastans的初始值爲0
0<n,m<=2*10^5
 

解題思路:

今天專門學了一下可持久化並查集,發現其實就是隻有底層結點的主席樹。

感覺也不用多解釋,有主席樹基礎的話代碼應該很好看懂。

但是這裏我不知道怎麼進行並查集的路徑壓縮,我採用了判斷秩大小的優化,聽隊友說這個可以把複雜度優化到O(logn),正常的路徑壓縮是可以優化到O(1)的應該。

具體的實現部分可以看代碼

 

Ac代碼:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
const int INF=1e9+7;
int n,m,cnt,root[maxn];
struct node
{
    int l,r;
    int val,dep;
}t[maxn*40];
void build(int &x,int l,int r)  //初始化
{
    x=++cnt;
    if(l==r)
    {
        t[x].val=l;
        return ;
    }
    int mid=(l+r)>>1;
    build(t[x].l,l,mid);
    build(t[x].r,mid+1,r);
}
void update(int &x,int y,int l,int r,int pos,int val)   //合併並查集
{
    t[++cnt]=t[y],x=cnt;
    if(l==r)
    {
        t[x].val=val;
        return ;
    }
    int mid=(l+r)>>1;
    if(pos<=mid) update(t[x].l,t[y].l,l,mid,pos,val);
    if(pos>mid) update(t[x].r,t[y].r,mid+1,r,pos,val);
}
int query(int x,int l,int r,int pos)    //查詢pos結點的值是否爲pos
{
    if(l==r) return x;
    int mid=(l+r)>>1;
    if(pos<=mid) return query(t[x].l,l,mid,pos);
    if(pos>mid) return query(t[x].r,mid+1,r,pos);
}
int find(int x,int val) //找尋父結點所在的位置
{
    int p=query(x,1,n,val);
    if(t[p].val==val) return p;
    return find(x,t[p].val);
}
void add(int x,int l,int r,int pos) //秩++
{
    if(l==r)
    {
        t[x].dep++;
        return ;
    }
    int mid=(l+r)>>1;
    if(pos<=mid) add(t[x].l,l,mid,pos);
    if(pos>mid) add(t[x].r,mid+1,r,pos);
}
int main()
{
    int ans=0;
    scanf("%d%d",&n,&m);
    build(root[0],1,n); //以root[0]爲初始狀態
    for(int i=1;i<=m;i++)
    {
        int flag,a,b;
        scanf("%d",&flag);
        if(flag==1)
        {
            scanf("%d%d",&a,&b);
            a=a^ans,b=b^ans;root[i]=root[i-1];  //將上個結點的值賦過來
            int fx=find(root[i],a),fy=find(root[i],b);  //找到fx fy對應的位置 注意 位置
            if(t[fx].val==t[fy].val) continue;  //相同跳過
            if(t[fx].dep>t[fy].dep) swap(fx,fy);    //根據秩的大小優化合並
            update(root[i],root[i-1],1,n,t[fx].val,t[fy].val);  //合併兩個並查集
            if(t[fx].dep==t[fy].dep) add(root[i],1,n,t[fy].val);    //秩++
        }
        else if(flag==2)
        {
            scanf("%d",&a); a=a^ans;
            root[i]=root[a];    //回溯版本
        }
        else 
        {
            scanf("%d%d",&a,&b);
            a=a^ans,b=b^ans;root[i]=root[i-1];
            int fx=find(root[i],a),fy=find(root[i],b);
            if(t[fx].val==t[fy].val) ans=1,printf("1\n");   //判斷是否屬於同一並查集
            else ans=0,printf("0\n");
        }
    }
    //system("pause");
}

 

發佈了146 篇原創文章 · 獲贊 14 · 訪問量 3萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章