bzoj 2588: Spoj 10628. Count on a tree（树上主席树）

2588: Spoj 10628. Count on a tree

Time Limit: 12 Sec   Memory Limit: 128 MB
Submit: 3980   Solved: 932
[ 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

暴力自重。。。

Source

鸣谢seter

[ Submit][ Status][ Discuss] 题解：树上主席树

大体思路就是对树上的每个点权进行离散化，然后对树上的的每一个节点到根的路径建一棵权值线段树，因为这些树形态结构都是相似的所有可以进行加减运算（主席树特性）。那么x,y两点之间的路径就可以表示成root[x]+root[y]-root[lca(x,y)]-root[fa[lca(x,y)],然后进行求解即可。

遇到的问题：直接对每一个节点建立的权值线段树，没有按照DFS序建树，于是RE了。为什么呢？因为在建树时是动态开点，也就是在他父亲所在的树上新开了一些节点，如果直接建树，有可能父亲节点还没有建，当前点就建立了。果然我还是太水了。。。。

#include
#include
#include
#include
#include
#define N 100003
#define M 6000000
using namespace std;
int a1[N],p[N],b1[N],n,m;
int v[N*2],point[N],next[N*2],tot;
int fa[N][20],mi[20],deep[N],sz,cnt,num[N],pos[N];
int ls[M],rs[M],sum[M],root[N],nm,pd[N];
int cmp(int x,int y)
{
	return a1[x]>i)&1)
	  x=fa[x][i];
	if (x==y) return x;
	for (int i=17;i>=0;i--)
	 if (fa[x][i]!=fa[y][i])
	  {
	  	x=fa[x][i];
	  	y=fa[y][i];
	  } 
	return fa[x][0];
}
int query(int x,int y,int k)
{
	int c=lca(x,y); int d=fa[c][0];
	int l=1; int r=cnt;
	int a=root[pos[x]]; int b=root[pos[y]]; c=root[pos[c]]; d=root[pos[d]];
	while (l