标签：Distance maxx 剖分 int top tree include root id

  边权转点权，每次遍历到下一个点，把走个这条边的权值加入主席树中即可。

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<string.h>
using namespace std;
const int maxx = 2e5+10;
struct node{
  int l,r,cnt;
}tree[maxx*40];
int head[maxx],rk[maxx],siz[maxx],top[maxx],son[maxx],d[maxx],fa[maxx],id[maxx],rt[maxx];
int dis[maxx];
struct Edge{
  int Next,to,w;
}edge[maxx<<1];
int root[maxx];
int tot=0,cnt=0,order;
void inserts(int l,int r,int pre,int &now,int pos){
    now=++cnt;
    tree[now]=tree[pre];
    tree[now].cnt++;
    if (l==r){
        return ;
    }
    int mid=(l+r)>>1;
    if (pos<=mid){
        inserts(l,mid,tree[pre].l,tree[now].l,pos);
    }else {
        inserts(mid+1,r,tree[pre].r,tree[now].r,pos);
    }
}
int query(int L,int R,int l,int r,int w){
    if (l==r){
        return tree[R].cnt-tree[L].cnt;
    }
    int mid=(l+r)>>1;
    if (w<=mid){
        return query(tree[L].l,tree[R].l,l,mid,w);
    }else {
        return tree[tree[R].l].cnt-tree[tree[L].l].cnt+query(tree[L].r,tree[R].r,mid+1,r,w);
    }
}
void add(int x,int y,int z){
  edge[++tot].to=y;
  edge[tot].w=z;
  edge[tot].Next=head[x];
  head[x]=tot;
}
void dfs1(int u,int f,int depth){
    d[u]=depth;
    fa[u]=f;
    siz[u]=1;
    for (int i=head[u];i;i=edge[i].Next){
        int v=edge[i].to;
        if (v==f)continue;
        dfs1(v,u,depth+1);
        dis[v]=edge[i].w;
        siz[u]+=siz[v];
        if(siz[v]>siz[son[u]])
            son[u]=v;
    }
}
void dfs2(int u,int t){
   top[u]=t;
   id[u]=++order;
   rk[order]=u;
  // cout<<dis[u]<<endl;
   inserts(0,1e9,root[order-1],root[order],dis[u]);
   if(!son[u])return;
   dfs2(son[u],t);
   for (int i=head[u];i;i=edge[i].Next)
   {
       int v=edge[i].to;
       if(v!=son[u] && v!=fa[u])
        dfs2(v,v);
   }
}
int query_line(int a,int b,int c){
  int ans=0;
  while(top[a]!=top[b]){
    if (d[top[a]]>d[top[b]])swap(a,b);
    ans+=query(root[id[top[b]]-1],root[id[b]],0,1e9,c);
    b=fa[top[b]];
  }
  if (d[a]>d[b])swap(a,b);
  ans+=query(root[id[a]],root[id[b]],0,1e9,c);
  return ans;
}
int main(){
  int w,op,n,uu,vv;
  scanf("%d%d",&n,&op);
  tot=0;
  cnt=0;
  order=0;
  for (int i=1;i<n;i++){
     scanf("%d%d%d",&uu,&vv,&w);
     add(uu,vv,w);
     add(vv,uu,w);
  }
  dfs1(1,0,1);
  dfs2(1,1);
//  cout<<endl;
//  for (int i=1;i<=n;i++){
//    cout<<dis[i]<<"--";
//  }
//  cout<<endl;
//  cout<<endl;
//  for (int i=1;i<=n;i++){
//    cout<<"--"<<id[i]<<endl;
//  }
  while(op--){
    scanf("%d%d%d",&uu,&vv,&w);
    printf("%d\n",w?query_line(uu,vv,w):0);
  }
  return 0;
}

标签：Distance,maxx,剖分,int,top,tree,include,root,id
来源： https://www.cnblogs.com/bluefly-hrbust/p/11397301.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。