【bzoj2588】Count on a tree 主席树

AC通道：http://www.lydsy.com/JudgeOnline/problem.php?id=2588

【题解】

我们用主席树维护每一个结点到根的路径（按dfs序建主席树即可），然后有一个优美的性质：

x到y的路径可以表示为T(x)+T(y)-T(t)-T(f[t])，其中t为lca(x,y)

这样在回答询问时处理这4个版本的线段树就行了。

经此题实测：手写hash离散比stl快500ms，果断选择手写hash，代码中被注释掉的部分是stl的离散代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<ctime>
#include<algorithm>
using namespace std;
#define FILE "read"
#define MAXN 100010
#define up(i,j,n) for(int i=j;i<=n;++i)
#define dn(i,j,n) for(int i=j;i>=n;--i)
#define cmax(a,b) a=max(a,b)
#define cmin(a,b) a=min(a,b)
namespace INIT{
	char buf[1<<15],*fs,*ft;
	inline char getc(){return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;}
	inline int read(){
		int x=0,f=1;  char ch=getc();
		while(!isdigit(ch))  {if(ch=='-')  f=-1;  ch=getc();}
		while(isdigit(ch))  {x=x*10+ch-'0';  ch=getc();}
		return x*f;
 	}
}using namespace INIT;
struct node{int y,next;}e[MAXN*2];
int n,m,ans(0),len,dfs_clock,tot,cnt,dfn[MAXN],Link[MAXN],deep[MAXN],root[MAXN],f[MAXN],hash[MAXN],a[MAXN],num[MAXN],sum[MAXN<<5],anc[MAXN][25],son[MAXN<<5][2];
void insert(int x,int y){e[++len].next=Link[x];Link[x]=len;e[len].y=y;}
void dfs(int x){
	dfn[++dfs_clock]=x; anc[x][0]=f[x];
	up(i,1,20)  anc[x][i]=anc[anc[x][i-1]][i-1];
	for(int i=Link[x];i;i=e[i].next){
		if(e[i].y==f[x])  continue;
		f[e[i].y]=x; deep[e[i].y]=deep[x]+1; 
		dfs(e[i].y);
	}
}
int find(int x){
	int l=1,r=tot;
	while(l+1<r){
		int mid=(l+r)>>1;
		if(hash[mid]<=x)  l=mid;
		else r=mid;
	}
	return hash[l]==x?l:r;
}
int lca(int x,int y){
	if(deep[x]<deep[y])  swap(x,y);
	dn(i,20,0)  if(deep[anc[x][i]]>=deep[y])  x=anc[x][i];
	if(x==y)  return x;
	dn(i,20,0)  if(anc[x][i]!=anc[y][i])  x=anc[x][i],y=anc[y][i];
	return f[x];
}
int newroot(int last){
	sum[++cnt]=sum[last]+1;
	son[cnt][0]=son[last][0]; 
	son[cnt][1]=son[last][1];
	return cnt;
}
void updata(int l,int r,int &root,int last,int x){
	root=newroot(last);
	if(l==r)  return;  int mid=(l+r)>>1;
	if(x<=mid)  updata(l,mid,son[root][0],son[last][0],x);
	else updata(mid+1,r,son[root][1],son[last][1],x);
}
int ask(int l,int r,int x,int y,int t,int tt,int k){
	if(l==r)  return hash[l];
	//if(l==r)  return num[l];
	int mid=(l+r)>>1;
	int temp=sum[son[x][0]]+sum[son[y][0]]-sum[son[t][0]]-sum[son[tt][0]];
	if(k<=temp)  return ask(l,mid,son[x][0],son[y][0],son[t][0],son[tt][0],k);
	else return ask(mid+1,r,son[x][1],son[y][1],son[t][1],son[tt][1],k-temp);
}
int main(){
	freopen(FILE".in","r",stdin);
	freopen(FILE".out","w",stdout);
	n=read();  m=read();  deep[1]=1;
	up(i,1,n)  a[i]=num[i]=read();
	sort(num+1,num+n+1);  hash[++tot]=num[1];
	up(i,2,n)  if(num[i]!=num[i-1])  hash[++tot]=num[i];
	//tot=unique(num+1,num+n+1)-(num+1);
	//up(i,1,n)  a[i]=lower_bound(num+1,num+tot+1,a[i])-num;
	up(i,1,n-1)  {int x=read(),y=read();insert(x,y);insert(y,x);}
	dfs(1);  
	up(i,1,n)  updata(1,tot,root[dfn[i]],root[f[dfn[i]]],find(a[dfn[i]]));
	//up(i,1,n)  updata(1,tot,root[dfn[i]],root[f[dfn[i]]],a[dfn[i]]);
	up(i,1,m){
		int x=read()^ans,y=read(),k=read(),t=lca(x,y);
		ans=ask(1,tot,root[x],root[y],root[t],root[f[t]],k);
		printf("%d",ans);  if(i!=m) printf("\n");
	}
	return 0;
}