POJ 1741

【题意】给了一颗树，问树上任意两个点瞒住dis(u,v)<=k的点对数。在树分治的论文上看到这个题，了解了什么叫做树的重心，老老实实学会了树的重心改怎么求，至于基于树的重心的点分治理解了很久，感觉还是一知半解。爱，要加强训练的强度了，再过段时间就是网络赛了，因此，更要加油了，先把这个球重心和基本的树分治放在这里！

【AC代码】

#include <stdio.h>
#include <string.h>
#include <assert.h>
#include <iostream>
#include <algorithm>
using namespace std;
//Tree Divide And Conquer.
//focus is root.
const int maxn = 11111;
const int maxm = 55555;
struct Edge{
   int v,w,next;
   Edge(){}
   Edge(int v,int w,int next):v(v),w(w),next(next){}
}E[maxm];
int tot,head[maxn];
int n,k,vis[maxn],ans,num,root;

void Init()
{
    ans=0,tot=0;
    memset(vis,0,sizeof(vis));
    memset(head,-1,sizeof(head));
}
void Add_Edge(int u,int v,int w)
{
    E[tot].v=v,E[tot].w=w,E[tot].next=head[u],head[u]=tot++;
}
// 重心 最大的最小
int mx[maxn],siz[maxn],mi,dis[maxn];
void dfssize(int u,int fa)//get size.

{
    siz[u]=1,mx[u]=0;
    for(int i=head[u]; ~i; i=E[i].next){
        int v = E[i].v;
        if(v!=fa&&!vis[v])
        {
            dfssize(v,u);
            siz[u]+=siz[v];
            if(siz[v]>mx[u]) mx[u] = siz[v];
        }
    }
}
void dfsroot(int r,int u,int fa)//求重心
{
    if(siz[r]-siz[u]>mx[u]) mx[u] = siz[r]-siz[u];
    if(mx[u]<mi) root=u,mi=mx[u];
    for(int i=head[u]; ~i; i=E[i].next){
        int v=E[i].v;
        if(v!=fa&&!vis[v])
        {
            dfsroot(r,v,u);
        }
    }
}
void dfsdis(int u,int fa,int d)//求距离
{
    dis[num++]=d;
    for(int i=head[u]; ~i; i=E[i].next){
        int v=E[i].v;
        if(v!=fa&&!vis[v]){
            dfsdis(v,u,d+E[i].w);
        }
    }
}
int cal(int u,int d)
{
    int sum=0;
    num=0;
    dfsdis(u,0,d);
    sort(dis,dis+num);
    int i=0,j=num-1;
    while(i<j)
    {
        while(dis[i]+dis[j]>k&&i<j) j--;
        sum+=j-i;
        i++;
    }
    return sum;
}
void dfs(int u)
{
    mi=n;
    dfssize(u,0);
    dfsroot(u,u,0);
    ans += cal(root,0);
    vis[root] = 1;
    for(int i=head[root]; ~i; i=E[i].next){//head[root]
        int v=E[i].v;
        if(!vis[v]){
            ans-=cal(v,E[i].w);
            dfs(v);
        }
    }
}
int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        if(n==0&&k==0)break;
        Init();
        int u,v,w;
        for(int i=1; i<=n-1; i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            Add_Edge(u,v,w);
            Add_Edge(v,u,w);
        }
        dfs(1);
        printf("%d\n",ans);
    }
    return 0;
}