洛谷 P3302 [SDOI2013]森林 主席樹+啓發式搜索

click here~:https://www.luogu.org/problem/P3302

emmm這個題是真的煩 一看題發現這題不是count on a tree的升級版麼

如果一點思路沒有的話還是先做那個題比較好

這個題多了一個條件就是說,你在查詢過程中需要連邊,仔細一樣腦海裏並沒有任何解題思路emmm,什麼lct根本就不會

這道題其實是用了個啓發式合併

何謂啓發式合併呢,我看了別人的代碼大概理解了一下:

題目上說了加邊過後還是樹結構,也就是說無論如何不會出現環什麼奇怪的東西。

你肯定還是遵循求lca拿來這條鏈在主席樹上搞個第k大

但是邊更新了之後呢,你的lca或者倍增數組都發生了變化

所以這個時候你要重新構建倍增數組,重新暴力的更新主席樹(其實也還真挺暴力)

因爲樹的邊數就是固定的v-1,所以一開始邊沒有給全,你的樹是分成一塊一塊的(一個聯通塊我們叫他小樹)。

如果你加邊,你需要吧這些小樹塊聯合起來,最暴力的想法,我就拿其中一個小樹連接到另一個小樹的上邊

就是一個是父節點,另一個就是子節點這樣的dfs,就相當於把一個塊加到另一個上邊了。

其實就相當與重新吧某一塊的點又遍歷了一遍

但是如果僅僅是暴力的吧某個加到另一個上肯定會爆炸,所以啓發式就是記錄每一個塊的個數,讓小樹中數量少的加到數量大的小樹上邊去,記錄個數就用一個數組並查集來維護。

說白了:就是n個節點,m個塊,你最後把他們連接成一顆完整的樹,你每一次吧小的小樹加到大的上,也就是在大的塊上再次更新小塊,自己琢磨琢磨,其實並沒有多暴力~~挺優雅~

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<set>
#include<stack>
#include<vector>
#include<map>
#include<queue>

#define myself i,l,r
#define lson i<<1
#define rson i<<1|1
#define Lson i<<1,l,mid
#define Rson i<<1|1,mid+1,r
#define half (l+r)/2
#define inff 0x3f3f3f3f
#define lowbit(x) x&(-x)
#define PI 3.14159265358979323846
#define min4(a, b, c, d) min(min(a,b),min(c,d))
#define min3(x, y, z) min(min(x,y),min(y,z))
#define pii make_pair
#define pr pair<int,int>
const int dir[4][2] = {0, -1, -1, 0, 0, 1, 1, 0};
typedef long long ll;
const ll inFF = 9223372036854775807;
typedef unsigned long long ull;
using namespace std;
const int maxn=1e5+5;
int fa[maxn];
int f[maxn][17],d[maxn],son[maxn];
int a[maxn],b[maxn],len;
int head[maxn<<1],sign;
int root[maxn],t;
int vis[maxn];
struct node
{
    int to,p;
}edge[maxn<<1];
struct nod
{
    int l,r,cnt;
}tr[maxn*400];
int n,m,q,lastans;
char s[2];
int find(int x)
{
    return (fa[x]==x)?x:fa[x]=find(fa[x]);
}
void add(int u,int v)
{
    edge[sign]=node{v,head[u]};
    head[u]=sign++;
}
int Hash(int x)
{
    return lower_bound(b+1,b+1+len,x)-b;
}
void insert(int &now,int pre,int l,int r,int k)
{
    now=++t;
    tr[now]=tr[pre];
    tr[now].cnt++;
    if(l==r) return;
    int mid=half;
    if(k<=mid) insert(tr[now].l,tr[pre].l,l,mid,k);
    else insert(tr[now].r,tr[pre].r,mid+1,r,k);
}
int query(int x1,int x2,int y1,int y2,int l,int r,int k)
{
    int mid=half;
    if(l==r) return b[l];
    int x=tr[tr[y1].l].cnt+tr[tr[y2].l].cnt-tr[tr[x1].l].cnt-tr[tr[x2].l].cnt;
    if(k<=x) return query(tr[x1].l,tr[x2].l,tr[y1].l,tr[y2].l,l,mid,k);
    else return query(tr[x1].r,tr[x2].r,tr[y1].r,tr[y2].r,mid+1,r,k-x);
}
void dfs(int u,int pre,int rt)
{
    f[u][0]=pre;
    for(int i=1;(1<<i)<=n;i++)
        f[u][i]=f[f[u][i-1]][i-1];
    d[u]=d[pre]+1;
    son[rt]++;
    fa[u]=pre;
    vis[u]=1;
    insert(root[u],root[pre],1,len,Hash(a[u]));
    for(int i=head[u];~i;i=edge[i].p)
    {
        int v=edge[i].to;
        if(v!=pre) dfs(v,u,rt);
    }
}
int get_lca(int x,int y)
{
    if(d[x]<d[y]) swap(x,y);
    int c=d[x]-d[y];
    for(int i=0;(1<<i)<=c;i++)
        if((1<<i)&c) x=f[x][i];
    if(x!=y)
    {
        for(int i=(int)log2(n);i>=0;i--)
            if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
        x=f[x][0];
    }
    return x;
}
void init()
{
    for(int i=0;i<=n;i++) head[i]=-1;
    sign=t=lastans=0;
}
int main()
{
    int c,x,y,k;
    cin>>c;
    cin>>n>>m>>q;
    init();
    for(int i=1;i<=n;i++) scanf("%d",&a[i]),b[i]=a[i],fa[i]=i;
    sort(b+1,b+1+n);
    len=unique(b+1,b+1+n)-b-1;
    for(int i=1;i<=m;i++)
        scanf("%d %d",&x,&y),add(x,y),add(y,x);
    for(int i=1;i<=n;i++)
    {
        if(!vis[i])
        {
            dfs(i,0,i);
            fa[i]=i;
        }
    }
    while(q--)
    {
        scanf("%s",s);
        if(s[0]=='Q')
        {
            scanf("%d %d %d",&x,&y,&k);
            x=x^lastans,y=y^lastans,k=k^lastans;
            int lca=get_lca(x,y);
            lastans=query(root[lca],root[f[lca][0]],root[x],root[y],1,len,k);
            cout<<lastans<<endl;
        }
        else
        {
            scanf("%d %d",&x,&y);
            x^=lastans,y^=lastans;
            add(x,y),add(y,x);
            int u=find(x),v=find(y);
            if(son[u]<son[v])//把大樹找出來,把小樹重新暴力加到大樹上,這裏算是優雅的暴力
            {
                swap(u,v);
                swap(x,y);
            }
            dfs(y,x,u);
        }
    }
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章