2588: Spoj 10628. Count on a tree

竟然1A了，完了这个月的RP用光了。

树上主席树（听起来好奇怪）。

在每个节点建一颗主席树，保存其到根节点的权值。

然后询问就是sum(u)+sum(v)-sum(lca(u,v))-sum(fa(lca(u,v)))，至于这个前缀和是什么，大概就是线性序列的那个前缀差不多，不过搞成根节点到当前节点链上的前缀和了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
inline int read(){
	int x=0,f=1;char ch;
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
const int N=100010;
struct Node{
	int l,r,w;
}tr[N*20];
int F[N*2],dep[N*2],st[N*2][20],cnt,pos[N],n;
int root[N],sz,dfn[N],dfs_clock,fa[N],rk[N];
int a[N],b[N],rank[N];
struct Edge{int to,next;}e[N*2];
int head[N],edge;
void ins(int u,int v){
	edge++;e[edge].to=v;e[edge].next=head[u];head[u]=edge;
}
inline bool cmp(int i,int j){
	return a[i]<a[j];
}
void dfs(int u,int depth){
	F[++cnt]=u;dep[cnt]=depth;pos[u]=cnt;st[cnt][0]=cnt;
	dfn[u]=++dfs_clock;rk[dfs_clock]=u;
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].to;
		if(v==fa[u])continue;
		fa[v]=u;
		dfs(v,depth+1);
		F[++cnt]=u;dep[cnt]=depth;st[cnt][0]=cnt;
	}
}
void rmq_init(){
	for(int j=1;(1<<j)<2*n;j++)
	for(int i=1;i+(1<<j)-1<2*n;i++){
		int l=st[i][j-1],r=st[i+(1<<(j-1))][j-1];
		if(dep[l]<dep[r])st[i][j]=l;
		else st[i][j]=r;
	}
}
int rmq(int l,int r){
	int k=0;
	while(1<<(k+1)<=r-l+1)k++;
	l=st[l][k];r=st[r-(1<<k)+1][k];
	if(dep[l]<dep[r])return l;
	return r;
}
int lca(int u,int v){
	int l=pos[u],r=pos[v];
	if(l>r)swap(l,r);
	return F[rmq(l,r)];
}
void update(int &i,int l,int r,int x){
	tr[++sz]=tr[i];i=sz;
	tr[i].w++;
	if(l==r)return;
	int mid=l+r>>1;
	if(x<=mid)update(tr[i].l,l,mid,x);
	else update(tr[i].r,mid+1,r,x);
}
int query(int a,int b,int A,int B,int l,int r,int k){
	if(l==r)return l;
	int tmp=tr[tr[a].l].w+tr[tr[b].l].w-tr[tr[A].l].w-tr[tr[B].l].w;
	int mid=l+r>>1;
	if(k<=tmp)return query(tr[a].l,tr[b].l,tr[A].l,tr[B].l,l,mid,k);
	else return query(tr[a].r,tr[b].r,tr[A].r,tr[B].r,mid+1,r,k-tmp);
}
int main(){
	int m;scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%d",&a[i]),rank[i]=i;
	sort(rank+1,rank+1+n,cmp);
	for(int i=1;i<=n;i++)b[rank[i]]=i;
	int u,v;
	for(int i=1;i<n;i++){
		scanf("%d%d",&u,&v);
		ins(u,v);ins(v,u);
	}
	int rt=1+n>>1;
	dfs(rt,0);
	rmq_init();
	for(int i=1;i<=n;i++){
		int t=rk[i];
		root[i]=root[dfn[fa[t]]];
		update(root[i],1,n,b[t]);
	}
	int k,last=0;
	for(int i=1;i<=m;i++){
		scanf("%d%d%d",&u,&v,&k);
		u^=last;
		int w=lca(u,v);
		last=a[rank[query(root[dfn[u]],root[dfn[v]],root[dfn[w]],root[dfn[fa[w]]],1,n,k)]];
		printf("%d",last);
		if(i!=m)printf("\n");
	}
	return 0;
}
		

为什么我的代码都这么短（斜眼）