poj 1741 - 点分治

解题思路：

以某一点为根节点的任意两点形成的路径有两种情况:

1.两个点的公共最近父祖先是根节点。

2.两个点的公共最近父祖先不是根节点。

这里记dis[i]为i节点到根节点的距离，那么不管两个节点是属于上面哪一种情况，只要满足dis[j]+dis[i]<=m（限制距离）,那么他们两个形成的路径一定是满足条件的，但是如果是第二个情况，虽然可以，但是白白加上了2倍的公共边，所以应该要减去这样的情况，那么这样的情况就应该在子问题上解决，防止重复。

重心:以某一节点为根节点，使得他的子树最大的那颗值最小，那么这个节点就被称为重心

当以该重心为根节点的树情况考虑完后就可以把它去掉了。

以重心为根节点的树进行分治大大减少了时间复杂度，递归深度不会大于log(n),那么我们又需要对dis进行排序nlog(n),所以时间复杂为O（nlog^2（n））

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const int mx = 1e4 + 10;
int head[mx],tot,n,m,ans,root,cnt;
int siz[mx],ma[mx],height,dis[mx];
bool vis[mx];
struct node
{
	int son,nxt;
	int v;	
}Edge[mx<<1];
void AddEdge(int x,int y,int v)
{
	Edge[tot].son = y;
	Edge[tot].v = v;
	Edge[tot].nxt = head[x];
	head[x] = tot++;
}
void dfssize(int x,int f)//求树大小 
{
	siz[x] = 1,ma[x] = 0;
	for(int i=head[x];~i;i=Edge[i].nxt)
	{
		int son = Edge[i].son;
		if(son==f||vis[son]) continue;
		dfssize(son,x);
		siz[x] += siz[son]; 
		ma[x] = max(siz[son],ma[x]);
	}
}
void dfsroot(int u,int x,int f)//求重心 
{
	ma[x] = max(ma[x],siz[u]-siz[x]);//如果x做重心还有另一颗子树就是根树减去x做子树
	if(ma[x]<height) height = ma[x],root = x;
	for(int i=head[x];~i;i=Edge[i].nxt)
	{
		int son = Edge[i].son;
		if(son==f||vis[son]) continue;
		dfsroot(u,son,x);
	}
	
}
void dfsdis(int x,int f,int d)//求距离 
{
	dis[cnt++] = d;
	for(int i=head[x];~i;i=Edge[i].nxt)
	{
		int son = Edge[i].son;
		if(son==f||vis[son]) continue;
		dfsdis(son,x,d+Edge[i].v);
	}
}
int get_sum(int x,int d)
{
	int ret = 0;cnt = 0;
	dfsdis(x,0,d);
	sort(dis,dis+cnt);
	int l = 0,r = cnt-1;
	while(l<r){
		while(dis[l]+dis[r]>m&&l<r) r--;
		ret += r - l;
		l++;
	}
	return ret;
}
void dfs(int x)
{
	height = n;
	dfssize(x,0);
	dfsroot(x,x,0);
	ans += get_sum(root,0);
	vis[root] = 1;
	for(int i=head[root];~i;i=Edge[i].nxt)
	{
		int son = Edge[i].son;
		if(!vis[son]){
			ans -= get_sum(son,Edge[i].v);
			dfs(son);
		}
	}
}
int main()
{
	while(scanf("%d%d",&n,&m)&&n+m){
		int a,b,c;
		memset(vis,0,sizeof(vis));
		memset(head,-1,sizeof(head));
		root = ans = tot = 0;
		for(int i=1;i<n;i++){
			scanf("%d%d%d",&a,&b,&c);
			AddEdge(a,b,c);
			AddEdge(b,a,c);
		}
		dfs(1);
		printf("%d\n",ans);
	}
	return 0;
}