3551: [ONTAK2010]Peaks加强版 kruskal重构树+dfs序+主席树

我的内心是崩溃的
尼玛bzoj题面样例都是同3545,然后我做完就顺手交了3545,然后就谜之RE调了一晚上,调了一晚上才突然发现!!!!!

这个题还是比较神的,膜拜popoqqq大爷
我们用点表示边,并且与这条边所连的两个点所属集合的顶部连边(并查集维护),于是就得到了一棵树。
这样得到的一棵树有一些性质:

1.二叉树(好吧这题意义不大)
2.原树与新树两点间路径上边权(点权)的最大值相等
3.子节点的边权小于等于父亲节点(大根堆)
4.原树中两点之间路径上边权的最大值等于新树上两点的LCA的点权

然后我们每次找到深度最小的点权 <=x <script type="math/tex" id="MathJax-Element-24"><=x</script>的点,求子树第 k 大直接按dfs序建主席树就行了。

#include<iostream>
#include<cstdio>
#include<algorithm>
#define N 200005
using namespace std;
int n,m,Q,cnt,tot,dfn,ans,size;
int h[N],v[N],f[N],num[N],root[N],deep[N],in[N],out[N],pos[N],hash[N],head[N],next[N],list[N];
int fa[N][20],mx[N][20];
int ls[5000005],rs[5000005],sum[5000005];
struct node {int x,y,z;} e[500005];
inline int read()
{
    int a=0,f=1; char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
    return a*f;
}
inline void insert(int x,int y)
{
    next[++cnt]=head[x];
    head[x]=cnt;
    list[cnt]=y;
}
inline bool operator<(node a,node b)
{
    return a.z<b.z;
}
int find(int i)
{
    return f[i]==i?i:f[i]=find(f[i]);
}
inline int Find(int x)
{
    int l=1,r=tot;
    while (l<=r)
    {
        int mid=l+r>>1;
        if (hash[mid]==x) return mid;
        else if (hash[mid]<x) l=mid+1;
        else r=mid-1;
    }
}
inline int top(int x,int up)
{
    for (int i=19;~i;i--)
        if ((1<<i)<=deep[x]&&mx[x][i]<=up) x=fa[x][i];
    return x;
}
void dfs(int x)
{
    in[x]=++dfn; pos[dfn]=x;
    for (int i=1;(1<<i)<=deep[x];i++)
        fa[x][i]=fa[fa[x][i-1]][i-1],mx[x][i]=max(mx[x][i-1],mx[fa[x][i-1]][i-1]);
    for (int i=head[x];i;i=next[i])
    {
        deep[list[i]]=deep[x]+1;
        fa[list[i]][0]=x;
        mx[list[i]][0]=v[x];
        dfs(list[i]);
    }
    out[x]=dfn;
}
void update(int l,int r,int x,int &y,int val)
{
    y=++size;
    sum[y]=sum[x]+1;
    if (l==r) return;
    ls[y]=ls[x]; rs[y]=rs[x]; 
    int mid=l+r>>1;
    if (val<=mid) update(l,mid,ls[x],ls[y],val);
    else update(mid+1,r,rs[x],rs[y],val);
}
inline int query(int x,int y,int k)
{
    int l=1,r=tot;
    if (k>sum[y]-sum[x]) return -1;
    while (l!=r)
    {
        int mid=l+r>>1;
        int tmp=sum[rs[y]]-sum[rs[x]];
        if (tmp>=k) l=mid+1,x=rs[x],y=rs[y];
        else r=mid,x=ls[x],y=ls[y],k-=tmp;
    }
    return l;
}
int main()
{
    n=read(); m=read(); Q=read();
    for (int i=1;i<=n;i++) h[i]=num[i]=read();
    sort(num+1,num+n+1);
    hash[++tot]=num[1];
    for (int i=2;i<=n;i++)
        if (num[i]!=num[i-1]) hash[++tot]=num[i];
    for (int i=1;i<=2*n-1;i++) f[i]=i;
    for (int i=1;i<=m;i++)
        e[i].x=read(),e[i].y=read(),e[i].z=read();
    sort(e+1,e+m+1);
    for (int i=1;i<=m;i++)
    {
        int p=find(e[i].x),q=find(e[i].y);
        if (p!=q)
        {
            n++;
            f[p]=f[q]=n; v[n]=e[i].z;
            insert(n,p); insert(n,q);
        }
    }
    dfs(n);
    for (int i=1;i<=n;i++)
        if (h[pos[i]]) update(1,tot,root[i-1],root[i],Find(h[pos[i]])); else root[i]=root[i-1];
    for (int i=1;i<=Q;i++)
    {
        int st=read(),up=read(),k=read();
        st^=ans,up^=ans,k^=ans;
        int t=top(st,up);
        if (sum[root[out[t]]]-sum[root[in[t]]]<k) ans=0,puts("-1"); 
        else printf("%d\n",ans=hash[query(root[in[t]],root[out[t]],k)]);
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值