[BZOJ3123][SDOI2013]森林(主席树合并)

题目:

我是超链接

题解:

一个是链上求第k大,另一个是连接一条边,强制在线
这样不是连通块内的第k大我们考虑权值线段树,连接一条边就是线段树的合并,对于链上的问题,我们需要采用可持久化的线段树-主席树,而且是树上主席树,即给每个点建一棵主席树,维护到根节点的所有点,这样答案就是sum(a)+sum(b)−sum(lca(a,b))−sum(fa(lca(a,b)))
还要连接一条边的话就是主席树的合并,合并要注意的就是合并之后原来的爸爸可能变成了儿子,即父子关系会发生变化,为了避免混乱,f数组要全部清空;要使用启发式合并,size小的放到size大的里
还要小技巧就是一棵树的size可以用并查集的根节点来维护,这样不用每次都更新

代码:

#include <map>
#include <cstdio>
#include <algorithm>
using namespace std;
const int sz=18;
const int N=80005;
int mi[sz+5],f[N][sz+5],tot,nxt[N*2],point[N],s,v[N*2],sum[N*200],ls[N*200],rs[N*200],h[N],zs,root[N];
int size[N],a[N],real[N],fa[N];map<int,int>has;
bool vis[N];
void addline(int x,int y)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
    ++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
}
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,int fa,int id)
{
    vis[x]=1;h[x]=h[fa]+1;
    root[x]=root[fa];if (id==0) size[x]=1;
    insert(root[x],1,s,a[x]);
    for (int i=1;i<sz;i++) f[x][i]=f[f[x][i-1]][i-1];
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa) 
      {
        f[v[i]][0]=x,dfs(v[i],x,id);
        if (id==0) size[x]+=size[v[i]];
      }
}
int lca(int x,int y)
{
    if (h[x]<h[y]) swap(x,y);
    int k=h[x]-h[y];
    for (int i=0;i<sz;i++)
      if (k>>i&1) x=f[x][i];
    if (x==y) return x;
    for (int i=sz-1;i>=0;i--)
      if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}
int qurry(int l,int r,int k,int x,int y,int old,int oldf)
{
    int mid=(l+r)>>1;
    if (l==r) return l;
    int tmp=sum[ls[x]]+sum[ls[y]]-sum[ls[old]]-sum[ls[oldf]];
    if (tmp>=k) return qurry(l,mid,k,ls[x],ls[y],ls[old],ls[oldf]);
    else return qurry(mid+1,r,k-tmp,rs[x],rs[y],rs[old],rs[oldf]);
}
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 T,n,m,q;scanf("%d",&T);
    scanf("%d%d%d",&n,&m,&q);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]),h[i]=a[i],fa[i]=i;
    sort(h+1,h+n+1);
    s=unique(h+1,h+n+1)-h-1;
    for (int i=1;i<=s;i++) has[h[i]]=i,real[i]=h[i];
    for (int i=1;i<=n;i++) a[i]=has[a[i]];
    for (int i=1;i<=m;i++)
    {
        int x,y;scanf("%d%d",&x,&y);
        addline(x,y);x=find(x);y=find(y);
        fa[x]=y;
    }
    for (int i=1;i<=n;i++)
      if (!vis[i]) dfs(i,0,0);
    int ans=0;
    while (q--)
    {
        int x,y,k;char st[5];scanf("%s",st);
        if (st[0]=='L')
        {
            scanf("%d%d",&x,&y); x^=ans; y^=ans;
            int fx=find(x),fy=find(y);
            if (size[fx]>size[fy]) swap(x,y),swap(fx,fy);
            f[x][0]=y; dfs(x,y,1); addline(x,y);
            size[fy]+=size[fx];fa[fx]=fy;
        }else
        {
            scanf("%d%d%d",&x,&y,&k);x^=ans; y^=ans; k^=ans;
            int old=lca(x,y),oldf=f[old][0];
            ans=real[qurry(1,s,k,root[x],root[y],root[old],root[oldf])];
            printf("%d\n",ans);
        }
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值