[BZOJ3551][ONTAK2010]Peaks加强版(kruskal+主席树)

题目:

我是超链接

题解:

强制在线的话我们就不能使用平衡树+启发式合并来做了
我们考虑做kruskal,每次合并两个连通块的时候创造一个新节点z,使fa[x]=fa[y]=z,这个点的权值就是边权了,我们可以这样建立出一棵树
不难发现,树上两个点的lca代表着图上两个点路径的最大值,并且因为kruskal,这里的最大值一定是最优的
另外,这棵树是一个大根堆
那么我们每次询问的时候从x点往上倍增找到可以承受的最大值,以这个节点为根节点的子树就是所有在范围内的节点,这个时候我们可以用主席树找区间第k大,只需要维护dfs序就可以了
注意是第k大还是第k小。。。
因为这里不会有双向边,所以深度h要在循环的时候添加,不然会RE。

代码:

#include <map>
#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
const int N=200100;
const int sz=18;
struct hh{int x,y,z;}li[N*3];
int f[N][20],mi[20],tot,nxt[N],point[N],v[N],zs,sum[N*25],ls[N*25],rs[N*25],dfn[N*3],h[N],maxx[N][20];
int num,zz[N],fa[N],a[N],b[N],root[N*3],in[N],out[N],n;bool vis[N];
int read()
{
    int x=0;char ch=getchar();
    while (ch<'0' || ch>'9') ch=getchar();
    while (ch<='9' && ch>='0') x=x*10+ch-'0',ch=getchar();
    return x;
}
void addline(int x,int y){++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;}
void insert(int &now,int l,int r,int x)
{
    sum[++zs]=sum[now]+1; ls[zs]=ls[now]; rs[zs]=rs[now]; now=zs;
    if (l==r) return;int mid=(l+r)>>1;
    if (x<=mid) insert(ls[now],l,mid,x);
    else insert(rs[now],mid+1,r,x);
}
void dfs(int x)
{
    vis[x]=1;dfn[++num]=x;
    for (int i=1;i<sz;i++)
      if (h[x]<mi[i]) break;
      else f[x][i]=f[f[x][i-1]][i-1],maxx[x][i]=max(maxx[x][i-1],maxx[f[x][i-1]][i-1]);
    for (int i=point[x];i;i=nxt[i])
      {
        f[v[i]][0]=x;maxx[v[i]][0]=zz[x];
        h[v[i]]=h[x]+1;dfs(v[i]);
      }
    if (x>n) dfn[++num]=x;
}
int beng(int now,int x)//从now开始蹦,蹦到不大于x的地方 
{
    for (int i=sz-1;i>=0;i--)
      if (h[now]>=mi[i] && maxx[now][i]<=x) now=f[now][i];
    return now;
}
int qurry(int l,int r,int x,int y,int k)
{
    if (l==r) return l;
    int mid=(l+r)>>1;
    int tmp=sum[rs[y]]-sum[rs[x]];
    if (tmp>=k) return qurry(mid+1,r,rs[x],rs[y],k);
    else return qurry(l,mid,ls[x],ls[y],k-tmp);
}
int cmp(hh a,hh b){return a.z<b.z;}
int find(int x){if (fa[x]!=x) fa[x]=find(fa[x]);return fa[x];}
int main()
{
    mi[0]=1;for (int i=1;i<sz;i++) mi[i]=mi[i-1]*2;
    int m,q;n=read();m=read();q=read();
    for (int i=1;i<=n;i++) a[i]=read(),b[i]=a[i];
    sort(b+1,b+n+1);
    for (int i=1;i<=n;i++) a[i]=lower_bound(b+1,b+n+1,a[i])-b;
    for (int i=1;i<=m;i++) li[i].x=read(),li[i].y=read(),li[i].z=read();
    sort(li+1,li+m+1,cmp);int neww=n;
    for (int i=1;i<=2*n;i++) fa[i]=i;
    for (int i=1;i<=m;i++)
    {
        int x=find(li[i].x),y=find(li[i].y);
        if (x!=y)
        {
            ++neww;zz[neww]=li[i].z;addline(neww,x);addline(neww,y);
            fa[x]=fa[y]=neww;
            num++;if (num==n-1) break;
        }
    }
    num=0;
    for (int i=1;i<=n;i++) if (!vis[i]) dfs(find(i));
    for (int i=1;i<=num;i++)
      if (dfn[i]<=n) 
      {
        root[i]=root[i-1];
        insert(root[i],1,n,a[dfn[i]]);
      }else
      {
        root[i]=root[i-1];
        if (!in[dfn[i]]) in[dfn[i]]=i;else out[dfn[i]]=i;
      }
    int ans=0;
    while (q--)
    {
        int v,x,k;v=read();x=read();k=read();
        if (ans!=-1) v^=ans,x^=ans,k^=ans;
        int t=beng(v,x);
        if (sum[root[out[t]]]-sum[root[in[t]]]>=k) ans=b[qurry(1,n,root[in[t]],root[out[t]],k)];
        else ans=-1;
        printf("%d\n",ans);
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值