【算法Camp】【每日一题】Namomo Spring Camp 2022 Div1 第 1 周 第 3 题

题目：
给出 n 个点的一棵树，每个点有各自的点权，多次询问两个点简单路径所构成点集的异或和。

输入格式
第一行两个数字 n 和 m , n 表示点数，m 表示询问次数 。

接下来一行 n 个整数 a1,a2,…,an ，表示每个点的点权。

接下来 n−1 行 , 每行两个整数 u,v ，表示点 u 和点 v 之间存在一条边。

再接下来 m 行，每行两个整数 u,v ，表示询问点 u 到点 v 的简单路径所构成点集的异或和。

输出格式
输出 m 行，对于每个询问，输出一行。

样例输入

7 3
0 1 2 3 4 5 6
1 2
1 3
2 4
2 5
3 6
3 7
4 6
4 7
5 6

样例输出

5
6
2

数据规模

所有数据保证 1≤n,m≤200000,1≤ai≤106。

题解：先用dfs处理出从根节点到每一个节点的异或和f[i]，然后每次查询两个点x，y 之间的异或和即为f[x] ^ f[y] ^ lca(x,y).
代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int N=400010;
int f[N];
int h[N],e[N],w[N],ne[N],idx;
int depth[N],fa[N][50];
void add(int a,int b){
	e[idx]=b;ne[idx]=h[a];h[a]=idx++;
}
void dfs(int u,int father){
	for(int i=h[u];i!=-1;i=ne[i])
	{
		int j=e[i];
		f[u]=f[father]^w[u];
		if(father==j)continue;
		dfs(j,u);
	}
}
void bfs(int root){
	memset(depth,0x3f,sizeof(depth));
	depth[root]=1;depth[0]=0;
	queue<int>q;
	q.push(root);
	while(q.size())
	{
		int t=q.front();
		q.pop();
		for(int i=h[t];i!=-1;i=ne[i])
		{
			int j=e[i];
			if(depth[j]>depth[t]+1)
			{
				depth[j]=depth[t]+1;
				q.push(j);
				fa[j][0]=t;
				for(int k=1;k<=40;k++)fa[j][k]=fa[fa[j][k-1]][k-1];
			}
		}
	}
}
int lca(int a,int b){
	if(depth[a]<depth[b])swap(a,b);
	for(int k=40;k>=0;k--)
	{
		if(depth[fa[a][k]]>=depth[b])a=fa[a][k];
	}
	if(a==b)return a;
	for(int k=40;k>=0;k--)
	{
		if(fa[a][k]!=fa[b][k])
		{
			a=fa[a][k];
			b=fa[b][k];
		}
	}
	return fa[a][0];
}
int main(){
	memset(h,-1,sizeof(h));
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%d",&w[i]);
	for(int i=1;i<n;i++)
	{
		int a,b;
		scanf("%d%d",&a,&b);
		add(a,b);add(b,a);
	}
	dfs(1,0);
	bfs(1);
	//printf("p: %d\n",f[2]^w[4]);
	for(int i=1;i<=m;i++)
	{
		int a,b;
		scanf("%d%d",&a,&b);
		int u=lca(a,b);
		int res=f[a]^f[b]^w[u];
		printf("%d\n",res);
	}
}