洛谷 2633 BZOJ 2588 Spoj 10628. Count on a tree-CSDN博客

【题解】

　　蜜汁强制在线。。。

　　每个点开一个从它到根的可持久化权值线段树。查询的时候利用差分的思想在树上左右横跳就好了。

 1 #include<cstdio>
 2 #include<algorithm>
 3 #define N 100010
 4 #define rg register
 5 #define ls (a[u].l)
 6 #define rs (a[u].r)
 7 using namespace std;
 8 int n,n2,m,tot,root[N],last[N],dep[N],top[N],hvy[N],fa[N],size[N],val[N],b[N];
 9 struct tree{
10     int sum,l,r;
11 }a[N*80];
12 struct edge{
13     int to,pre;
14 }e[N<<1];
15 inline int read(){
16     int k=0,f=1; char c=getchar();
17     while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
18     while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
19     return k*f;
20 }
21 inline int qlca(int x,int y){
22     int f1=top[x],f2=top[y];
23     while(f1!=f2){
24         if(dep[f1]<dep[f2]) swap(x,y),swap(f1,f2);
25         x=fa[f1]; f1=top[x];
26     }
27     return dep[x]<dep[y]?x:y;
28 }
29 void update(int &u,int l,int r,int pos){
30     a[++tot]=a[u]; a[u=tot].sum++;
31     if(l==r) return;
32     int mid=(l+r)>>1;
33     if(pos<=mid) update(ls,l,mid,pos);
34     else update(rs,mid+1,r,pos);
35 }
36 int query(int u,int v,int lca,int f,int l,int r,int k){
37     if(l==r) return l;
38     int tmp=a[ls].sum+a[a[v].l].sum-a[a[lca].l].sum-a[a[f].l].sum,mid=(l+r)>>1;
39     return k<=tmp?query(ls,a[v].l,a[lca].l,a[f].l,l,mid,k):query(rs,a[v].r,a[lca].r,a[f].r,mid+1,r,k-tmp);
40 }
41 void dfs1(int x){
42     size[x]=1; dep[x]=dep[fa[x]]+1;
43     root[x]=root[fa[x]]; update(root[x],1,n2,val[x]);
44     for(rg int i=last[x],to;i;i=e[i].pre)if((to=e[i].to)!=fa[x]){
45         fa[to]=x; dfs1(to);
46         size[x]+=size[to];
47         if(size[to]>size[hvy[x]]) hvy[x]=to;
48     }
49 }
50 void dfs2(int x,int tp){
51     top[x]=tp;
52     if(hvy[x]) dfs2(hvy[x],tp);
53     for(rg int i=last[x],to;i;i=e[i].pre)
54         if((to=e[i].to)!=fa[x]&&to!=hvy[x]) dfs2(to,to);
55 }
56 int main(){
57     n=read(); m=read();
58     for(rg int i=1;i<=n;i++) val[i]=b[i]=read();
59     sort(b+1,b+1+n); n2=unique(b+1,b+1+n)-b-1;
60     for(rg int i=1;i<=n;i++) val[i]=lower_bound(b+1,b+1+n2,val[i])-b;
61     for(rg int i=1;i<n;i++){
62         int u=read(),v=read();
63         e[++tot]=(edge){v,last[u]}; last[u]=tot;
64         e[++tot]=(edge){u,last[v]}; last[v]=tot;
65     }
66     tot=0; dfs1(1); dfs2(1,1);
67     int last=0;
68     while(m--){
69         int u=read()^last,v=read(),k=read(),l=qlca(u,v),f=fa[l];
70         printf("%d\n",last=b[query(root[u],root[v],root[l],root[f],1,n2,k)]);
71     }
72     return 0;
73 }