[bzoj2733][HNOI2012]永无乡

最新推荐文章于 2019-02-27 21:00:31 发布

YZH__12345

最新推荐文章于 2019-02-27 21:00:31 发布

阅读量309

点赞数 1

分类专栏：线段树合并

本文链接：https://blog.csdn.net/YZHYZHYZH3/article/details/78900335

版权

线段树合并专栏收录该内容

2 篇文章 0 订阅

订阅专栏

Orz w_yqts
线段树合并裸题
每合并一次减少一个节点,所以复杂度可以接受

#include <bits/stdc++.h>
using namespace std;
#define N 100005
struct node
{
    int s,l,r;
}tr[2333333];
int f[N],rt[N],a[N],id[N];
int n,trsz;
inline int find(int x)
{
    if (f[x]==x) return x;
    return f[x]=find(f[x]);
}
void ins(int &k,int l,int r,int t)
{
    if (!k) k=++trsz;
    ++tr[k].s;
    if (l==r) return;
    int mid=(l+r)>>1;
    if (t<=mid) ins(tr[k].l,l,mid,t);else ins(tr[k].r,mid+1,r,t);
}
inline int merge(int x,int y)
{
    if (!x) return y;
    if (!y) return x;
    if (!tr[x].l && !tr[x].r && !tr[y].l && !tr[y].r)
    {
        tr[x].s=tr[x].s+tr[y].s;
        return x;
    }
    tr[x].l=merge(tr[x].l,tr[y].l);
    tr[x].r=merge(tr[x].r,tr[y].r);
    tr[x].s=tr[tr[x].l].s+tr[tr[x].r].s;
    return x;
}
int query(int k,int l,int r,int t)
{
    if (l==r) return l;
    int mid=(l+r)>>1;
    if (t<=tr[tr[k].l].s) return query(tr[k].l,l,mid,t);
    return query(tr[k].r,mid+1,r,t-tr[tr[k].l].s);
}
inline void solve()
{
    char s[10];
    int x,y;
    scanf("%s%d%d",s,&x,&y);
    x=find(x);
    if (s[0]=='B') {y=find(y);if (x!=y) rt[x]=merge(rt[x],rt[y]);f[y]=x;return;}
    if (tr[rt[x]].s<y) {puts("-1");return;}
    int t=query(rt[x],1,n,y);
    printf("%d\n",id[t]);
}
int main()
{
    int Q,x,y;
    cin>>n>>Q;
    for (int i=1;i<=n;++i) scanf("%d",&a[i]),f[i]=i;
    while (Q--) scanf("%d%d",&x,&y),f[find(y)]=find(x);
    for (int i=1;i<=n;++i)
    {
        ins(rt[find(i)],1,n,a[i]);
        id[a[i]]=i;
    }
    cin>>Q;
    while (Q--) solve();
    return 0;
}