POJ 1741 点分治

好久没更新我的博客了，最近跟着@orzGEOTCBRL和@GE_he学了点分治，就打了这道题作为练习题。

打这道题的时候还因为vector忘记清零而RE

还因为找重心写错变成了深搜TAT

#include <cstdio>
#include <vector>
#include <algorithm>
#include <cstring>
#define fi first
#define se second
using namespace std;
int d[40000],size[40000],tag[40000],f[40000];
int len,n,k,ans,xx,yy,zz;
vector < pair<int,int> > v[40000];
bool vis[40000];
void getdep(int x,int fa,int dep){
	d[++len]=dep;
	for (int i=0;i<v[x].size();i++) if (!vis[v[x][i].fi] && v[x][i].fi!=fa) getdep( v[x][i].fi,x,dep+v[x][i].se); 
}
void getsize(int x,int fa){
	size[x]=1;
	for (int i=0;i<v[x].size();i++) if (!vis[v[x][i].fi] && v[x][i].fi!=fa) getsize(v[x][i].fi,x),size[x]+=size[v[x][i].fi];
	tag[++tag[0]]=x;
}
void solve_d(int opt){
	sort(d+1,d+len+1);
	int l=1,r=len;
	while (l<r) if (d[l]+d[r]<=k) ans+=opt*(r-l),l++; else r--;
}
int getroot(int x){
	int maxsize=200000;
	int root=0;
	getsize(x,0);
	for (int i=1;i<=tag[0];i++){
		int mm=0;
		for (int j=0;j<v[tag[i]].size();j++){
			int u=v[tag[i]][j].fi;
			if (!vis[u] && mm<size[u]) mm=size[u];
		}
		f[tag[i]]=max(mm,size[x]-size[tag[i]]);
	}
	for (int i=1;i<=tag[0];i++) if (maxsize>f[tag[i]]) maxsize=f[tag[i]],root=tag[i];
	return root;
}
void solve(int x){
	tag[0]=len=0;
	int rt=getroot(x);
	getdep(rt,0,0);
	solve_d(1);
	vis[rt]=true;
	for (int i=0;i<v[rt].size();i++) if (!vis[v[rt][i].fi]){
		len=0;getdep(v[rt][i].fi,rt,v[rt][i].se);
		solve_d(-1);
	}
	for (int i=0;i<v[rt].size();i++) if (!vis[v[rt][i].fi]) solve(v[rt][i].fi);
}
int main(){
	while (scanf("%d%d",&n,&k)==2){
		ans=0;
		if (n==0 && k==0) break;
		memset(vis,false,sizeof(vis));
		for (int i=1;i<n;i++){
			scanf("%d%d%d",&xx,&yy,&zz);
			v[xx].push_back(make_pair(yy,zz));
			v[yy].push_back(make_pair(xx,zz));
		}
		solve(1);
		printf("%d\n",ans);
		for (int i=1;i<=n;i++) v[i].clear();
	}
	return 0;
}