luoguP1456 Monkey King

題目鏈接

分析

左偏樹的模板題,加上並查集之後還是很好理解的,沒有什麼思路可言就是合併和修改堆中的元素;
代碼+註釋

#include<bits/stdc++.h>
#define N 100005
using namespace std;
int n,x,y,k,h,m,cnt,fa[N],v[N],l[N],r[N],d[N];
int find(int x){return fa[x]==x ? x:fa[x]=find(fa[x]);}//模板並查集
int merge(int x,int y){
    if(!x || !y)return x+y;
    if(v[x]<v[y])swap(x,y);
    r[x]=merge(r[x],y);
    fa[r[x]]=x;
    if(d[l[x]]<d[r[x]])swap(l[x],r[x]);
    if(!r[x]) d[x]=0;else d[x]=d[r[x]]+1;
    return x;
}//合併堆,就基本操作
int del(int x){
    int left=l[x],right=r[x];//注意此處不可省略
    fa[left]=left;
    fa[right]=right;
    l[x]=r[x]=d[x]=0;
    return merge(left,right);//否則此處會出現問題,想一想爲什麼
}//刪除節點
void work(int x,int y){
    v[x]/=2,v[y]/=2;
    int left,right;
    left=del(x);
    right=del(y);//刪除原來的節點
    left=merge(x,left);
    right=merge(y,right);//跟新新的節點
    int h=merge(left,right);//合併
    printf("%d\n",v[h]);//輸出
}
int main(){
    while(~scanf("%d",&n)){
        for(int i=1;i<=n;i++){
            fa[i]=i;
            l[i]=r[i]=d[i]=0;
            scanf("%d",&v[i]);
        }//有多組數據所以需要初始化
        scanf("%d",&m); 
        for(int i=1;i<=m;i++){
            int x,y;
            scanf("%d%d",&x,&y);
            int fx=find(x),fy=find(y);//並查集查找父親節點
            if(fx==fy){
                printf("-1\n");
                continue;
            }
            work(fx,fy);
        }
    }
    return 0;
}

心得

對於此類題目需要細心細心再細心,數據結構不能錯,一錯就調三百年QwQ;

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