bzoj 3545: [ONTAK2010]Peaks

最新推荐文章于 2020-01-31 22:30:03 发布

fyc_kabuto

最新推荐文章于 2020-01-31 22:30:03 发布

阅读量198

点赞数 1

分类专栏：线段树合并

本文链接：https://blog.csdn.net/qq_36808030/article/details/79473960

版权

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

5 篇文章 0 订阅

订阅专栏

题意：

在Bytemountains有N座山峰，每座山峰有他的高度h_i。有些山峰之间有双向道路相连，共M条路径，每条路径有一个困难值，这个值越大表示越难走，现在有Q组询问，每组询问询问从点v开始只经过困难值小于等于x的路径所能到达的山峰中第k高的山峰，如果无解输出-1。

题解：

直接离线+线段树合并。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int inf=1e9;
int n,m,Q;
struct node{
    int x,y,c;
}e[500010];
struct trnode{
    int lc,rc,c;
}tr[4000000];int tot=0,root[100010];
int fa[100010],h[100010];
int findfa(int x) {return fa[x]==x?x:fa[x]=findfa(fa[x]);}
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
void update(int &x,int l,int r,int k)
{
    if(!x) x=++tot;
    if(l==r) {tr[x].c++;return;}
    int mid=(l+r)/2;
    if(k<=mid) update(tr[x].lc,l,mid,k);
    else update(tr[x].rc,mid+1,r,k);
    tr[x].c=tr[tr[x].lc].c+tr[tr[x].rc].c;
}
void merge(int &x1,int x2)
{
    if(!x1) {x1=x2;return;}
    if(!x2) return;
    tr[x1].c+=tr[x2].c;
    merge(tr[x1].lc,tr[x2].lc);
    merge(tr[x1].rc,tr[x2].rc);
}
struct query{
    int v,x,k,ans,num;
}q[500010];
bool cmp(node a,node b) {return a.c<b.c;}
bool cmpq(query a,query b) {return a.x<b.x;}
int findans(int x,int l,int r,int k)
{
    if(tr[x].c<k) return -1;
    if(l==r) return l;
    int mid=(l+r)/2,rcc=tr[tr[x].rc].c;
    if(rcc>=k) return findans(tr[x].rc,mid+1,r,k);
    else return findans(tr[x].lc,l,mid,k-rcc);
}
int main()
{
    n=read();m=read();Q=read();
    for(int i=1;i<=n;i++)
    {
        fa[i]=i;h[i]=read();
        update(root[i],1,inf,h[i]);
    }
    for(int i=1;i<=m;i++) e[i].x=read(),e[i].y=read(),e[i].c=read();
    sort(e+1,e+m+1,cmp);
    for(int i=1;i<=Q;i++) q[i].v=read(),q[i].x=read(),q[i].k=read(),q[i].num=i;
    sort(q+1,q+Q+1,cmpq);
    int p=0;
    for(int i=1;i<=Q;i++)
    {
        while(p<m&&e[p+1].c<=q[i].x)
        {
            p++;int x=e[p].x,y=e[p].y;
            x=findfa(x);y=findfa(y);
            if(x!=y) fa[x]=y,merge(root[y],root[x]);
        }
        q[q[i].num].ans=findans(root[findfa(q[i].v)],1,inf,q[i].k);
    }
    for(int i=1;i<=Q;i++) printf("%d\n",q[i].ans);
}