poj1741点分治

点分治一般按点分为两种情况：
1:过该点,则遍历统计答案
2:不过,递归到子树重心统计
然后合并答案

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
const int N=1e5+10,inf=(1<<30);
int last[N],dep[N],dis[N],s[N],vis[N],f[N],
n,K,len,root,sum,ans;
struct Edge{int to,next,val;Edge(int to=0,int next=0,int val=0):to(to),next(next),val(val){}}e[N<<1];
void add_edge(int u,int v,int w){e[++len]=Edge(v,last[u],w);last[u]=len;}
void getroot(int x,int fa)
{
    s[x]=1;f[x]=0;
    for(int i=last[x];i;i=e[i].next) {
        int id=e[i].to;
        if(id==fa||vis[id])continue;
        getroot(id,x);
        s[x]+=s[id];f[x]=max(f[x],s[id]);
    }                                                                               
    f[x]=max(f[x],sum-s[x]);
    if(f[x]<f[root])root=x;
}
void getdep(int x,int fa)
{
    dep[++dep[0]]=dis[x];
    for(int i=last[x];i;i=e[i].next) {
        int id=e[i].to;
        if(id==fa||vis[id])continue;
        dis[id]=dis[x]+e[i].val;
        getdep(id,x);
    }
}
int cal(int x,int w)
{
    dep[0]=0;dis[x]=w;
    getdep(x,0);
    sort(dep+1,dep+1+dep[0]);
    int l=1,r=dep[0],t=0;
    for(;l<r;) {
        if(dep[l]+dep[r]<=K){t+=(r-l);l++;}
        else r--;
    }
    return t;
}
void solve(int x)
{
    ans+=cal(x,0);vis[x]=1;
    for(int i=last[x];i;i=e[i].next) {
        int id=e[i].to;
        if(vis[id])continue;
        ans-=cal(id,e[i].val);
        sum=s[id];root=0;getroot(id,root);
        solve(root);
    }
}
int main()
{
    while(1) {
        root=len=ans=0;
        memset(vis,0,sizeof(vis));
        memset(last,0,sizeof(last));
        n=read();K=read();
        if(!n)break;
        for(int u,v,w,i=1;i<n;i++) {
            u=read();v=read();w=read();
            add_edge(u,v,w);add_edge(v,u,w);
        }
        sum=n;
        f[0]=inf;getroot(1,0);
        solve(root);
        printf("%d\n",ans);
    }
    return 0;
}