BZOJ BZOJ 2588: Spoj 10628. Count on a tree 树上主席树

2588: Spoj 10628. Count on a tree

Time Limit: 12 Sec  Memory Limit: 128 MB
Submit: 6970  Solved: 1686
[Submit][Status][Discuss]

Description

给定一棵N个节点的树，每个点有一个权值，对于M个询问(u,v,k)，你需要回答u xor lastans和v这两个节点间第K小的点权。其中lastans是上一个询问的答案，初始为0，即第一个询问的u是明文。

Input

第一行两个整数N,M。

第二行有N个整数，其中第i个整数表示点i的权值。

后面N-1行每行两个整数(x,y)，表示点x到点y有一条边。

最后M行每行两个整数(u,v,k)，表示一组询问。

Output

M行，表示每个询问的答案。最后一个询问不输出换行符

Sample Input

8 5
105 2 9 3 8 5 7 7
1 2
1 3
1 4
3 5
3 6
3 7
4 8
2 5 1
0 5 2
10 5 3
11 5 4
110 8 2

Sample Output

2
8
9
105
7

HINT

HINT：

N,M<=100000

暴力自重。。。

主席树维护该节点到root的权值线段树

区间加加减减就行了

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=100100;

int n,ans,lim,cnt;

int ref[N],V[N],root[N];

struct P{int val,pos;friend bool operator <(const P &x,const P &y){return x.val<y.val;}}p[N];

int ecnt,last[N];
struct EDGE{int to,nt;}e[N<<1];
inline void add(int u,int v)
{e[++ecnt]=(EDGE){v,last[u]};last[u]=ecnt;}

struct president_tree{int ls,rs,w;}tr[N*20];

void insert(int &k,int x,int l,int r,int val)
{
	k=++cnt;tr[k].w=tr[x].w+1;
	if(l==r)return ;tr[k].ls=tr[x].ls;tr[k].rs=tr[x].rs;
	int mid=(l+r)>>1;
	val<=mid?insert(tr[k].ls,tr[x].ls,l,mid,val):insert(tr[k].rs,tr[x].rs,mid+1,r,val);
}

int fa[N][18],d[N];

void dfs(int u)
{
	d[u]=d[fa[u][0]]+1;
	insert(root[u],root[fa[u][0]],1,lim,ref[u]);
	for(int i=last[u];i;i=e[i].nt)
	{
		if(fa[u][0]==e[i].to)continue;
		fa[e[i].to][0]=u;dfs(e[i].to);
	}
}

void RMQ_init()
{
	register int i,j;
	for(j=1;j<=17;++j)for(i=1;i<=n;++i)fa[i][j]=fa[fa[i][j-1]][j-1];
}

inline int getlca(int u,int v)
{
	if(d[u]<d[v])swap(u,v);
	int len=d[u]-d[v];
	for(int i=0;(1<<i)<=len;++i)if(len&(1<<i))u=fa[u][i];
	if(u==v)return u;
	for(int i=17;i>=0;--i)if(fa[u][i]^fa[v][i])u=fa[u][i],v=fa[v][i];
	return fa[u][0];
}

void query(int u,int v,int k)
{
	register int lca=getlca(u,v),falca=fa[lca][0],l=1,r=lim,sum=0,a=root[lca],b=root[falca],c=root[u],d=root[v];
	while(l<r)
	{
		sum=0;
		sum+=tr[tr[c].ls].w+tr[tr[d].ls].w;sum-=tr[tr[a].ls].w+tr[tr[b].ls].w;
		if(sum<k)
		{
			c=tr[c].rs;d=tr[d].rs;a=tr[a].rs;b=tr[b].rs;
			k-=sum;l=((l+r)>>1)+1;
		}
		else 
		{
			c=tr[c].ls;d=tr[d].ls;a=tr[a].ls;b=tr[b].ls;
			r=(l+r)>>1;
		}
	}
	ans=V[l];print(ans);
}

int main()
{
	n=read();int Q=read();
	register int i,u,v,k;
	for(i=1;i<=n;++i)p[i].val=read(),p[i].pos=i;
	sort(p+1,p+1+n);V[++lim]=p[1].val;ref[p[1].pos]=lim;
	for(i=2;i<=n;++i){if(p[i].val^V[lim])V[++lim]=p[i].val;ref[p[i].pos]=lim;}
	for(i=1;i<n;++i){u=read();v=read();add(u,v);add(v,u);}
	dfs(1);RMQ_init();
	while(Q--)
	{
		u=read()^ans;v=read();k=read();
		query(u,v,k);if(Q)puts("");
	}
	return 0;
}
/*
8 5
105 2 9 3 8 5 7 7
1 2
1 3
1 4
3 5
3 6
3 7
4 8
2 5 1
0 5 2
10 5 3
11 5 4
110 8 2

2
8
9
105
7
*/