POJ1741 Tree

题意：找出一棵树中有多少个点对之间的距离小于等于k。。

论文上的题。

点分治，找到树的重心，然后计算经过这个重心的点对，计算从根到这个子树每个节点的距离，然后找到所有点对过根的距离小于等于k的，然后再减去再同一棵子树里的就行了，然后分治一直递归下去。。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=10010;
const int INF=1<<30;
typedef long long ll;
bool vis[MAXN];
struct EDGE
{
	int v,next;
	int val;
}edge[MAXN<<1];
int head[MAXN],size;
ll S[MAXN];
void init()
{
	memset(head,-1,sizeof(head));
	size=0;
}
void add_edge(int u,int v,int c)
{
	edge[size].v=v;
	edge[size].val=c;
	edge[size].next=head[u];
	head[u]=size++;
}
int num[MAXN],siz[MAXN],root;
ll dep[MAXN],k;
int tot_size;
void get_root(int u,int fa)	//找树的重心
{
	siz[u]=1;
	num[u]=0;
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].v;
		if(vis[v]||v==fa)
			continue;
		get_root(v,u);
		siz[u]+=siz[v];
		num[u]=max(num[u],siz[v]);
	}
	num[u]=max(num[u],tot_size-num[u]);
	if(num[root]>num[u])
		root=u;
}
int cnt;
void get_dep(int u,int fa)	//每个点离重心的距离
{
	S[cnt++]=dep[u];
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].v;
		if(v==fa||vis[v])
			continue;
		dep[v]=dep[u]+edge[i].val;
		get_dep(v,u);
	}
}
ll get_num(int u,int fa,int len)	//计算
{
	cnt=0;
	dep[u]=len;
	get_dep(u,fa);
	sort(S,S+cnt);
	ll l=0,r=cnt-1;
	ll res=0;
	while(l<r)
	{
		if(S[r]+S[l]<=k)
		{
			res+=r-l;
			l++;
		}
		else
			r--;
	}
	return res;
}
ll ans;
void dfs(int u,int fa)
{
	ans+=get_num(u,fa,0);	//找到经过重心的路径
	vis[u]=1;
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].v;
		if(vis[v])
			continue;
		ans-=get_num(v,u,edge[i].val);	//减去不经过路径的重心
		root=0;
		tot_size=siz[v];
		memset(num,0,sizeof(num));
		num[0]=INF;
		get_root(v,-1);
		dfs(root,-1);
	}
}
int main()
{
	int n,i;
	while(scanf("%d%I64d",&n,&k)==2)
	{
		if(n==0&&k==0)
			break;
		int u,v,c;
		init();
		for(i=1;i<n;i++)
		{
			scanf("%d%d%d",&u,&v,&c);
			add_edge(u,v,c);
			add_edge(v,u,c);
		}
		memset(num,0,sizeof(num));
		memset(vis,0,sizeof(vis));
		num[0]=INF;
		tot_size=n;
		root=0;
		get_root(1,-1);
		ans=0;
		dfs(root,-1);
		printf("%I64d\n",ans);
	}
	return 0;
}