bzoj 3551: [ONTAK2010]Peaks加强版 克鲁斯卡尔重构树+主席树

题目大意:
给你一个 n n n个点 m m m条边的无向图,边有边权,点有高度。
q q q个询问,每个询问一个点 x x x在不经过边权大于 c c c的边能到达的点中高度第 k k k大的点的高度。

分析:
建出克鲁斯卡尔重构树。对于询问点 x x x,倍增求出深度最浅的一个权值小于等于 c c c的点,然后对子树求第 k k k大。可以用dfs序+可持久化线段树解决。

代码:

/**************************************************************
    Problem: 3551
    User: liangzihao
    Language: C++
    Result: Accepted
    Time:16272 ms
    Memory:95756 kb
****************************************************************/
 
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
 
const int N=200005;
 
int n,m,Q,val[N],f[N],last[N],fa[N][25],dfn[N],tim,tot,sz,mn[N],mx[N],cnt,a[N],root[N];
struct edge{int to,next,u,v,w;}e[N*10];
struct tree{int s,l,r;}t[N*10];
 
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;
}
 
bool cmp(edge a,edge b)
{
    return a.w<b.w;
}
 
void addedge(int u,int v)
{
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].next=last[v];last[v]=cnt;
}
 
int find(int x)
{
    if (f[x]==x) return x;
    else return f[x]=find(f[x]);
}
 
void dfs(int x)
{
    if (x<=n) mn[x]=mx[x]=++tim,dfn[tim]=x;
    else mn[x]=n+1;
    for (int i=1;i<=20;i++) fa[x][i]=fa[fa[x][i-1]][i-1];
    for (int i=last[x];i;i=e[i].next)
    {
        if (e[i].to==fa[x][0]) continue;
        fa[e[i].to][0]=x;
        dfs(e[i].to);
        mn[x]=min(mn[x],mn[e[i].to]);
        mx[x]=max(mx[x],mx[e[i].to]);
    }
}
 
void ins(int &d,int p,int l,int r,int x)
{
    d=++sz;t[d]=t[p];t[d].s++;
    if (l==r) return;
    int mid=(l+r)/2;
    if (x<=mid) ins(t[d].l,t[p].l,l,mid,x);
    else ins(t[d].r,t[p].r,mid+1,r,x);
}
 
int get(int x,int w)
{
    for (int i=20;i>=0;i--)
        if (fa[x][i]&&val[fa[x][i]]<=w) x=fa[x][i];
    return x;
}
 
int query(int d,int p,int l,int r,int x)
{
    if (t[d].s-t[p].s<x) return -1;
    if (l==r) return l;
    int mid=(l+r)/2;
    if (t[t[d].r].s-t[t[p].r].s>=x) return query(t[d].r,t[p].r,mid+1,r,x);
    else return query(t[d].l,t[p].l,l,mid,x-t[t[d].r].s+t[t[p].r].s);
}
 
int main()
{
    n=read();m=read();Q=read();
    for (int i=1;i<=n;i++) val[i]=read(),a[i]=val[i];
    for (int i=1;i<=n*2;i++) f[i]=i;
    sort(a+1,a+n+1);
    int a1=unique(a+1,a+n+1)-a-1;
    for (int i=1;i<=n;i++) val[i]=lower_bound(a+1,a+a1+1,val[i])-a;
    for (int i=1;i<=m;i++) e[i].u=read(),e[i].v=read(),e[i].w=read();
    sort(e+1,e+m+1,cmp);
    tot=n;
    for (int i=1;i<=m;i++)
    {
        int x=find(e[i].u),y=find(e[i].v);
        if (x!=y)
        {
            tot++;val[tot]=e[i].w;
            addedge(tot,x);addedge(tot,y);
            f[x]=f[y]=tot;
        }
    }
    dfs(tot);
    for (int i=1;i<=n;i++) ins(root[i],root[i-1],1,n,val[dfn[i]]);
    int ans=0;
    while (Q--)
    {
        int v=read(),x=read(),k=read();
        if (ans>-1) v^=ans,x^=ans,k^=ans;
        int p=get(v,x);ans=query(root[mx[p]],root[mn[p]-1],1,n,k);
        if (ans>-1) ans=a[ans];
        printf("%d\n",ans);
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值