Distance on the tree（数剖 + 主席树）

题目链接：https://nanti.jisuanke.com/t/38229

题目大意：给你n个点，n-1条边，然后是m次询问，每一次询问给你u,v,w然后问你从u -> v 的路径上有多少边是小于等于w的、

AC代码：

 #include<iostream>
 #include<cmath>
 #include<stack>
 #include<queue>
 #include<stdio.h>
 #include<string>
 #include<cstring>
 #include<algorithm>
 using namespace std;
 # define inf 0x3f3f3f3f
 # define ll long long
 const int maxn = 3e5+;
 int n,m,num,tot,cnt,totn;
 int a[maxn];
 int head[maxn];
 int root[maxn];
 struct Query
 {
     int l,r,tt;
 } que[maxn];
 struct Tree
 {
     int ls,rs,sum;
 } tr[maxn*];
 struct Edge
 {
     int from,to,val,s;
 } edges[maxn<<];
 void addedge(int x,int y,int z)
 {
     edges[++tot]=Edge{x,y,z,head[x]};
     head[x]=tot;
 }
 int d[maxn],fa[maxn],size[maxn],w[maxn];
 int son[maxn];
 int rk[maxn],kth[maxn],top[maxn];
 void dfs1(int u,int pre,int val)
 {

     d[u]=d[pre]+;
     fa[u]=pre;
     size[u]=;
     w[u]=val;
     for(int i=head[u]; i!=-; i=edges[i].s)
     {
         Edge &e=edges[i];
         if(e.to==pre)
             continue;
         dfs1(e.to,u,e.val);
         size[u]+=size[e.to];
         if(size[e.to]>size[son[u]])
             son[u]=e.to;
     }
 }
 void dfs2(int u,int y)
 {
     rk[u]=++cnt;
     kth[cnt]=u;
     top[u]=y;
     if(son[u]==)
         return ;
     dfs2(son[u],y);
     for(int i=head[u]; i!=-; i=edges[i].s)
     {
         Edge &e=edges[i];
         if(e.to==son[u]||e.to==fa[u])
             continue;
         dfs2(e.to,e.to);
     }
 }
 void buildtree(int &x,int l,int r)
 {
     x=++totn;
     if(l==r)
         return ;
     int mid=(l+r)>>;
     buildtree(tr[x].ls,l,mid);
     buildtree(tr[x].rs,mid+,r);
 }
 void add(int &x,int last,int l,int r,int p)
 {
     x=++totn;
     tr[x]=tr[last];
     if(l==r)
     {
         tr[x].sum++;
         return ;
     }
     int mid=l+r>>;
     if(p<=mid)
         add(tr[x].ls,tr[last].ls,l,mid,p);
     if(p> mid)
         add(tr[x].rs,tr[last].rs,mid+,r,p);
     tr[x].sum=tr[tr[x].ls].sum+tr[tr[x].rs].sum;
 }
 int ask(int ql,int qr,int l,int r,int kk)
 {
     if(<=l&&r<=kk)
         return tr[qr].sum-tr[ql].sum;
     int mid=l+r>>,ans=;
     if(<=mid)
         ans+=ask(tr[ql].ls,tr[qr].ls,l,mid,kk);
     if(mid<kk)
         ans+=ask(tr[ql].rs,tr[qr].rs,mid+,r,kk);
     return ans;
 }
 int get_sum(int x,int y,int tt)
 {
     int ans=;
     int fx=top[x],fy=top[y];
     while(fx!=fy)
     {
         if(d[fx]<d[fy])
             swap(x,y),swap(fx,fy);
         ans+=ask(root[rk[fx]-],root[rk[x]],,num,tt);
         x=fa[fx];
         fx=top[x];
     }
     if(d[x]<d[y])
         swap(x,y);
     ans+=ask(root[rk[y]],root[rk[x]],,num,tt);
     return ans;
 }
 int main()
 {
     scanf("%d %d",&n,&m);
     ll x,y,z,id,ans;
     for(int i=; i<=n; i++)
     {
         head[i]=-;
     }
     for(int i=; i<=n-; i++)
     {
         scanf("%d %d %d",&x,&y,&z);
         a[++num]=z;
         addedge(x,y,z);
         addedge(y,x,z);
     }
     for(int i=; i<=m; i++)
     {
         scanf("%d %d %d",&que[i].l,&que[i].r,&que[i].tt);
         a[++num]=que[i].tt;
     }
     sort(a+,a+num+);
     num=unique(a+,a+num+)-a-;
     dfs1(,,inf);
     dfs2(,);
     for(int i=; i<=n; i++)
     {
         w[i]=lower_bound(a+,a+num+,w[i])-a;
     }
     buildtree(root[],,num);
     for(int i=; i<=n; i++)
     {
         add(root[i],root[i-],,num,w[kth[i]]);
     }
     for(int i=; i<=m; i++)
     {
         que[i].tt=lower_bound(a+,a+num+,que[i].tt)-a;
         int ans=get_sum(que[i].l,que[i].r,que[i].tt);
         printf("%d\n",ans);
     }
     return ;
 }