题目描述
给定一棵N个节点的树,每个点有一个权值,对于M个询问(u,v,k),你需要回答u xor lastans和v这两个节点间第K小的点权。其中lastans是上一个询问的答案,初始为0,即第一个询问的u是明文。
输入格式
第一行两个整数N,M。
第二行有N个整数,其中第i个整数表示点i的权值。
后面N-1行每行两个整数(x,y),表示点x到点y有一条边。
最后M行每行两个整数(u,v,k),表示一组询问。
输出格式
M行,表示每个询问的答案。
输入输出样例
输入 #1 复制
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
输出 #1 复制
2
8
9
105
7
说明/提示
HINT:
N,M<=100000
解题思路:
还是在静态第K小,树上做主席树
可用u+v-lca-lca_fa得到权值个数,所以还得有LCA
#include <iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<cmath>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn = 1e5+5;
long long val[maxn],uiq[maxn];
int n, m, h, cnt;
struct Edge{
int to,nxt;
}edges[maxn<<1];
int head[maxn<<1];
int tot;
int dp[maxn][20];
int depth[maxn];
void addEdge(int u, int v)
{
edges[tot] = Edge{v,head[u]};
head[u] = tot++;
}
void init()
{
memset(head,-1,sizeof(head));
tot = 1;
sort(uiq+1,uiq+1+n);
h = unique(uiq+1,uiq+1+n)-uiq-1;
cnt = 0;
}
int id(long long val)
{
return lower_bound(uiq+1,uiq+1+n,val)-uiq;
}
int sum[maxn<<5],L[maxn<<5],R[maxn<<5],T[maxn];
int build(int l, int r)
{
int rt = ++cnt;
sum[rt] = 0;
if(l==r) return rt;
int mid = (l+r)>>1;
L[rt] = build(l,mid);
R[rt] = build(mid+1,r);
return rt;
}
int update(int pre, int l, int r, int x)
{
int rt = ++cnt;
L[rt] = L[pre];
R[rt] = R[pre];
int mid = (l+r)>>1;
sum[rt] = sum[pre]+1;
if(l<r)
{
if(x<=mid) L[rt] = update(L[pre],l,mid,x);
else R[rt] = update(R[pre],mid+1,r,x);
}
return rt;
}
void dfs(int u, int fa)
{
depth[u] = depth[fa]+1;
dp[u][0] = fa;
T[u] = update(T[fa],1,h,id(val[u]));
for(int i = 1; i < 20; i++)
{
dp[u][i] = dp[dp[u][i-1]][i-1];
}
for(int i = head[u]; ~i; i = edges[i].nxt)
{
int v = edges[i].to;
if(v== fa)continue;
else dfs(v,u);
}
}
int lca(int a, int b)
{
if(depth[a]>depth[b]) swap(a,b);
for(int i = 19; i >= 0; i--)
{
if(depth[a]<depth[b]&&depth[a]<=depth[dp[b][i]])
b = dp[b][i];
}
if(a==b)
return a;
for(int i = 19; i >= 0; i--)
{
if(dp[a][i]!=dp[b][i])
{
a = dp[a][i];
b = dp[b][i];
}
}
return dp[a][0];
}
int query(int t1, int t2, int t3, int t4, int k, int l, int r)
{
if(l==r)
return l;
int mid = (l+r)>>1;
int tmp = sum[L[t1]]+sum[L[t2]]-sum[L[t3]]-sum[L[t4]];
// cout <<l<<" "<<r<<" "<<tmp <<" "<<sum[L[t1]]<<" "<<sum[L[t2]]<<" "<<sum[L[t3]]<<" "<<sum[L[t4]]<<endl;
if(tmp>=k)
return query(L[t1],L[t2],L[t3],L[t4],k,l,mid);
else
return query(R[t1],R[t2],R[t3],R[t4],k-tmp,mid+1,r);
}
int main()
{
scanf("%d%d", &n, &m);
for(int i = 1; i <= n; i++)
{
scanf("%lld", &val[i]);
uiq[i] = val[i];
}
init();
// cout <<"## ";
// for(int i = 1; i <= h; i++)
// cout << uiq[i]<<" ";
// cout <<endl;
// cout <<" && "<<id(3)<<endl;
int u, v, k;
for(int i = 1; i < n; i++)
{
scanf("%d%d", &u, &v);
addEdge(u,v);
addEdge(v,u);
}
T[0] = build(1,h);
dfs(1,0);
int ans = 0;
while(m--)
{
scanf("%d%d%d", &u, &v, &k);
u = u^ans;
int pre = lca(u,v);
// cout <<"lca "<<u<<" "<<v<<" "<<pre<<endl;
int pp = dp[pre][0];
int pos = query(T[u],T[v],T[pre],T[pp],k,1,h);
ans = uiq[pos];
printf("%d\n",ans);
}
return 0;
}