poj 1741 点分治

题意，一棵树，边有权值，问有多少点对之间的路径<=K 

思路，树分治基础，直接点分治，合并的时候分边依次计算。维护处理边的dis数组和某重心已经处理过的dis数组就好了，记得保证有序性方便二分计数。

代码：

//#include<bits/stdc++.h>
#include <cstring>
#include <algorithm>
#include <cstdio>
using namespace std;
#define X first
#define Y second
#define PB push_back
#define MP make_pair
#define MEM(a,b) memset(a,b,sizeof(a))
typedef long long ll;
const ll mod = 1e9+7;
const int maxn =1e6+10;
ll n,k;
int N,K;
int ans,root,Max;
struct node{
    int v,next,w;
}edge[maxn*2];
int head[maxn],tot,num1,num2;
int si[maxn],maxv[maxn],vis[maxn];//vis标记重心
int dis[maxn],tmp[maxn],tmp2[maxn];
void init(){
    tot=ans=0;MEM(head,-1);MEM(vis,0);
}
void add_edge(int u,int v,int w){
    edge[tot].v=v;edge[tot].w=w;edge[tot].next=head[u];head[u]=tot++;
}
//处理子树的大小（固定）
void dfssi(int u,int f){
    si[u]=1;
    maxv[u]=0;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].v;
        if(v==f||vis[v])continue;
        dfssi(v,u);
        si[u]+=si[v];
        if(si[v]>maxv[u])maxv[u]=si[v];
    }
}
//找重心（固定）
void dfsroot(int r,int u,int f){
    if(si[r]-si[u]>maxv[u])//si[r]-si[u]是u上面部分的树的尺寸，跟u的最大孩子比，找到最大孩子的最小差值节点
        maxv[u]=si[r]-si[u];
    if(maxv[u]<Max)Max=maxv[u],root=u;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].v;
        if(v==f||vis[v])continue;
        dfsroot(r,v,u);
    }
}

//求每个点到u的距离（自定义）
void dfsdis(int u,int d,int f){
    tmp[num2++]=d;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].v;
        if(v!=f&&!vis[v])//不跨过其他重心
            dfsdis(v,d+edge[i].w,u);
    }
}

//分边解决一个重心的答案（自定义）
int calc(int u,int d){
    int ret=0;
    num1=0;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].v,w=edge[i].w;
        if(!vis[v]){
            num2=0;
            dfsdis(v,w,0);
            sort(tmp,tmp+num2);
            ret+=(int)(upper_bound(tmp,tmp+num2,K)-tmp);
            for(int j=0;j<num2;j++) ret+=(int)(upper_bound(dis,dis+num1,K-tmp[j])-dis);
            merge(dis,dis+num1,tmp,tmp+num2,tmp2);
            copy(tmp2,tmp2+num1+num2,dis);
            num1+=num2;
        }
    }
    return ret;
}

void dfs(int u){
    Max=N;
    dfssi(u,0);//重心预处理
    dfsroot(u,u,0);//找重心
    vis[root]=1;
    ans+=calc(root,0);//合并贡献（固定？）
    for(int i=head[root];i!=-1;i=edge[i].next){
        int v=edge[i].v;
        if(!vis[v])dfs(v);
    }
}

int main(){
    while(scanf("%d%d",&N,&K),N||K){
        int u,v,w;
        init();
        for(int i=1;i<N;i++){
            scanf("%d%d%d",&u,&v,&w);
            add_edge(u,v,w);add_edge(v,u,w);
        }
        dfs(1);
        printf("%d
",ans);
    }
    return 0;
}