[BZOJ3545][ONTAK2010]Peaks(线段树合并||splay启发式合并)

题目:

我是超链接

题解:

首先是splay启发式合并,我们离线后将询问排序,对于每一个询问,我们把比ta小的边权加入然后给连通块维护splay,每次启发式合并查询排名
我们可以用更加优秀的线段树合并,不要忘了线段树也可以用来查询区间第k大,首先这个范围是1e9我们要离散化,然后和splay的启发式合并是一样的,只是我们每次合并线段树。时间复杂度 O(nlogn) O ( n l o g n )
说是nlogn怎么这么慢刚刚卡过去

代码:

#include <map>
#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
const int N=500005;
int size[N*4],ls[N*4],rs[N*4],sz,ans[N],f[N],h[N],a[N],root[N],real[N];
struct hh{int x,y,z,id;}lin[N],ask[N];
map<int,int> has;
int cmp(hh a,hh b){return a.z<b.z;}
int cmp1(hh a,hh b){return a.y<b.y;}
int findfa(int x){if (f[x]!=x) f[x]=findfa(f[x]);return f[x];}
int read()
{
    int x=0;char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}
void add(int &x,int l,int r,int w)
{
    if (!x) x=++sz; size[x]++;
    if (l==r) return;
    int mid=(l+r)>>1;
    if (w<=mid) add(ls[x],l,mid,w);
    else add(rs[x],mid+1,r,w);
}
int qurry(int now,int l,int r,int k)
{
    if (l==r) return l;
    int mid=(l+r)>>1;
    if (size[rs[now]]>=k) return qurry(rs[now],mid+1,r,k);
    else return qurry(ls[now],l,mid,k-size[rs[now]]);  
}
int merge(int x,int y)
{
    if (!x || !y) return x+y;
    if (!ls[x] && !rs[x]){size[x]+=size[y];return x;}
    ls[x]=merge(ls[x],ls[y]);
    rs[x]=merge(rs[x],rs[y]);
    size[x]=size[ls[x]]+size[rs[x]];
    return x;
}
int main()
{
    int n,m,q;n=read();m=read();q=read();
    for (int i=1;i<=n;i++) h[i]=read(),a[i]=h[i],f[i]=i;
    sort(a+1,a+n+1);
    int s=unique(a+1,a+n+1)-a-1;
    for (int i=1;i<=s;i++) has[a[i]]=i;
    for (int i=1;i<=n;i++) real[has[h[i]]]=h[i],h[i]=has[h[i]];
    for (int i=1;i<=n;i++) add(root[i],1,s,h[i]);
    for (int i=1;i<=m;i++) lin[i].x=read(),lin[i].y=read(),lin[i].z=read();
    sort(lin+1,lin+m+1,cmp);
    for (int i=1;i<=q;i++) ask[i].x=read(),ask[i].y=read(),ask[i].z=read(),ask[i].id=i;
    sort(ask+1,ask+q+1,cmp1);
    int last=1;
    for (int i=1;i<=q;i++)
    {
        for (;last<=m && lin[last].z<=ask[i].y;last++)
        {
            int x=lin[last].x,y=lin[last].y;
            int fx=findfa(x),fy=findfa(y);
            if (fx==fy) continue;
            root[fy]=merge(root[fx],root[fy]); f[fx]=fy;
        }
        int x=findfa(ask[i].x);
        if (size[root[x]]<ask[i].z) ans[ask[i].id]=-1;
        else ans[ask[i].id]=real[qurry(root[x],1,s,ask[i].z)];
    }
    for (int i=1;i<=q;i++) printf("%d\n",ans[i]);
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值