poj1741 tree 点分治

原题:http://poj.org/problem?id=1741

题解:求长度不超过k的路径长度。这是道点分治的题。点分治首先要求树的重心。然后从重心开始就出过重心的答案，在向子树转移减掉重复的。在重复上面的过程。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<vector>
#define N 10010 
#define inf 1<<30
using namespace std; 
struct E{int to,c,nxt;}data[N<<1];
int len=1,h[N],f[N],sz[N],vis[N],rt,n,k,ans;
vector<int> v; 

inline void ins(int x,int y,int c){
	data[++len].to=y;data[len].c=c;data[len].nxt=h[x];h[x]=len;
	data[++len].to=x;data[len].c=c;data[len].nxt=h[y];h[y]=len;
}
void getroot(int x,int fa){//求重心 
	sz[x]=1;f[x]=0;//sz表示子节点个数,f表示最大子树的节点数 
	for(int i=h[x];i;i=data[i].nxt){
		int y=data[i].to;if(vis[y] || y==fa) continue;// 
		getroot(y,x);
		sz[x]+=sz[y];f[x]=max(f[x],sz[y]);
	}
	f[x]=max(f[x],n-sz[x]);
	if(f[rt]>f[x]) rt=x;
}

void dfs(int x,int fa,int c){//求路径 
	for(int i=h[x];i;i=data[i].nxt){
		int y=data[i].to;if(fa==y || vis[y]) continue;
		dfs(y,x,c+data[i].c);
	}
	v.push_back(c);
}

int work(int x,int c){
	v.clear();dfs(x,0,c);
	sort(v.begin(),v.end());
	int l=0;int r=v.size()-1;int s=0;
	while(l<r){
		if(v[l]+v[r]<=k)s+=r-l,l++;
		else r--;
	}
	return s;
}
void sol(int x){
	ans+=work(x,0);
	vis[x]=1;
	for(int i=h[x];i;i=data[i].nxt){
		int y=data[i].to;if(vis[y]) continue;
		ans-=work(y,data[i].c);
		n=sz[y];rt=0;getroot(y,0);
		sol(rt);
	}
}
int main(){
	while(~scanf("%d%d",&n,&k)){
		if(n==0 && k==0) break;
		
		memset(vis,0,sizeof vis);
		memset(h,0,sizeof h);
		rt=0;f[rt]=inf;ans=0;len=1; 
		
		for(int i=1;i<n;i++){
			int x,y,c;
			scanf("%d%d%d",&x,&y,&c);
			ins(x,y,c);
		}
		getroot(1,0);
		sol(rt);
		printf("%d\n",ans);	
	}
	return 0;
}