标签:bzoj2588 spoj-10628 count-tree 可持久化线段树 lca
#include <stdio.h>
int main()
{
puts("转载请注明出处[vmurder]谢谢");
puts("网址:blog.csdn.net/vmurder/article/details/45048639");
}
对于每个树上节点存一个版本的可持久化线段树,为它到根节点上所有权值的权值线段树(需要离散化)。
然后对于每次询问,这条链(a,b)的线段树就是:
然后线段树上求第k小啦。
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 101000
#define LOGN 20
#define LS s[last].l
#define RS s[last].r
#define ls s[now].l
#define rs s[now].r
using namespace std;
struct LSH
{
int x,id;
void read(int i){id=i;scanf("%d",&x);}
bool operator < (const LSH &a)const
{return x<a.x;}
}lsh[N];
int n,m,p;
struct Eli
{
int v,next;
}e[N<<1];
int head[N],cnt,val[N];
inline void add(int u,int v)
{
e[++cnt].v=v;
e[cnt].next=head[u];
head[u]=cnt;
}
struct Segment_Tree
{
int l,r,x;
}s[N*LOGN];
int root[N];
void add(int last,int &now,int x,int L=1,int R=p)
{
now=++cnt;
if(L==R)s[now].x=s[last].x+1;
else {
int mid=L+R>>1;
if(x<=mid)add(LS,ls,x,L,mid),rs=RS;
else add(RS,rs,x,mid+1,R),ls=LS;
s[now].x=s[ls].x+s[rs].x;
}
}
int query(int a,int b,int A,int B,int k,int L=1,int R=p)
{
int x=s[s[a].l].x+s[s[b].l].x-s[s[A].l].x-s[s[B].l].x;
if(L==R)return L;
int mid=L+R>>1;
if(k<=x)return query(s[a].l,s[b].l,s[A].l,s[B].l,k,L,mid);
else return query(s[a].r,s[b].r,s[A].r,s[B].r,k-x,mid+1,R);
}
int fa[N][LOGN],dep[N];
void dfs(int x,int p)
{
fa[x][0]=p;
dep[x]=dep[p]+1;
add(root[p],root[x],val[x]);
int i,v;
for(i=head[x];i;i=e[i].next)
{
v=e[i].v;
if(v==p)continue;
dfs(v,x);
}
}
void array()
{
for(int k=1;k<LOGN;k++)
for(int i=1;i<=n;i++)
fa[i][k]=fa[fa[i][k-1]][k-1];
}
int getlca(int x,int y)
{
if(dep[x]<dep[y])swap(x,y);
for(int k=LOGN-1;~k;k--)
if(dep[fa[x][k]]>=dep[y])
x=fa[x][k];
if(x==y)return x;
for(int k=LOGN-1;~k;k--)
if(fa[x][k]!=fa[y][k])
x=fa[x][k],y=fa[y][k];
return fa[x][0];
}
int lastans;
int main()
{
int i,a,b,c;
scanf("%d%d",&n,&m);
for(i=1;i<=n;i++)lsh[i].read(i);
sort(lsh+1,lsh+n+1);
for(i=1;i<=n;i++)
{
if(i==1||lsh[i].x!=lsh[i-1].x)lsh[++p].x=lsh[i].x;
val[lsh[i].id]=p;
}
for(cnt=0,i=1;i<n;i++)
{
scanf("%d%d",&a,&b);
add(a,b),add(b,a);
}
cnt=0,dfs(1,0);
array();
while(m--)
{
scanf("%d%d%d",&a,&b,&c),a^=lastans;
i=getlca(a,b);
a=query(root[a],root[b],root[i],root[fa[i][0]],c);
lastans=lsh[a].x;
printf("%d\n",lastans);
}
return 0;
}
【BZOJ2588】【Spoj 10628.】 Count on a tree 可持久化线段树+lca
标签:bzoj2588 spoj-10628 count-tree 可持久化线段树 lca
原文地址:http://blog.csdn.net/vmurder/article/details/45048639