FZU 2237 中位数 主席树 树上k大

#include <cstdio>
#include <cstring>
#include <queue>
#include <set>
#include <stack>
#include <cstdlib>
#include <algorithm>
#include <time.h>
#include <vector>
#include <cmath>
using namespace std;
typedef long long LL;
typedef pair<int,int>pii;
const int N=5e4+;
const int S=1e5+;
const int INF=0x3f3f3f3f;
struct Edge{
   int v,w,next;
}edge[N*];
int head[N],tot,root[N],n,q;
void add(int u,int v,int w){
   edge[tot].v=v;
   edge[tot].w=w;
   edge[tot].next=head[u];
   head[u]=tot++;
}
struct Node{
   int l,r,v;
}o[N*];
int sz;
void add(int &rt,int l,int r,int x){
   o[++sz]=o[rt];rt=sz;
   ++o[rt].v;
   if(l==r)return;
   int m=(l+r)>>;
   if(x<=m)add(o[rt].l,l,m,x);
   else add(o[rt].r,m+,r,x);
}
int ask(int x,int y,int z,int l,int r,int k){
    if(l==r)return l;
    int c1=o[o[x].l].v+o[o[y].l].v-*o[o[z].l].v;
    int m=(l+r)>>;
    if(c1>=k)ask(o[x].l,o[y].l,o[z].l,l,m,k);
    else ask(o[x].r,o[y].r,o[z].r,m+,r,k-c1);
}
int fa[N][],d[N];
void dfs(int u,int f,int w){
   fa[u][]=f;d[u]=d[f]+;
   if(u!=)add(root[u]=root[f],,S,w);
   for(int i=head[u];~i;i=edge[i].next){
      int v=edge[i].v;
      if(v==f)continue;
      dfs(v,u,edge[i].w);
   }
}
int lca(int u,int v){
    if(d[u]<d[v])swap(u,v);
    for(int t=d[u]-d[v],i=;t;t>>=,++i)
      if(t&)u=fa[u][i];
    if(u==v)return u;
    for(int i=;i>=;--i){
      if(fa[u][i]!=-&&fa[u][i]!=fa[v][i])
        u=fa[u][i],v=fa[v][i];
    }
    return fa[u][];
}
int main(){
    while(~scanf("%d%d",&n,&q)){
       tot=sz=;
       memset(head,-,sizeof(head));
       for(int i=;i<n;++i){
          int u,v,w;
          scanf("%d%d%d",&u,&v,&w);
          ++w;
          add(u,v,w),add(v,u,w);
       }
       memset(fa,-,sizeof(fa));
       dfs(,,);
       for(int j=;j<=;++j){
          for(int i=;i<=n;++i){
             if(fa[i][j-]!=-)fa[i][j]=fa[fa[i][j-]][j-];
          }
       }
       while(q--){
          int u,v;
          scanf("%d%d",&u,&v);
          int tp=lca(u,v);
          int k=o[root[u]].v+o[root[v]].v-*o[root[tp]].v;
          if(k&)k=k/+;
          else k/=;
          printf("%d\n",ask(root[u],root[v],root[tp],,S,k)-);
       }
    }
    return ;
}