点分治入门---树中点对距离详（？）解

题目大意：给出一棵带边权的树，问有多少对点的距离<=Len

其中树最多有10000个点，Len<=maxlongint

---

这是一道模板题目。

对于一个点x，我们考虑如何求解经过x点的路径方案数，设为f（x）。

考虑直接从x出发做一遍DFS，算出所有未到过的点到x点的距离，然后排序。

设两个指针，如果length[l]+length[r]<=len,那么length[l]+length[x]（l<x<=r）合法且还未被统计过，ans+=r-l，左指针右移一格；否则右指针左移一格。

统计答案结束。

怎么可能结束了呢？这样计算肯定会有不经过x点的路径答案啊！

在计算f（son（son为x的儿子））时，由于x点已经经过，算答案时x点相当于断掉的开关，所以算出的答案路径一定不经过x，减去即可。

每次都挑重心去递归，这样每棵子树大小至多为原来的一半，只需递归N log N层

快排log N复杂度，总时间复杂度O（N log^2 N）

code

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int i,j,k,m,n,o,p,l,s,t,maxlen,tot,cent,ans;
int f[20005][4],q[20005],size[20005],bz[20005],son[20005];
long long len[20005];
void insert(int x,int y,int z) {f[++t][1]=y,f[t][2]=q[x],f[t][3]=z,q[x]=t;}
void getlen(int t,int fa,int lenn)
{
	len[++len[0]]=lenn;
	for (int k=q[t];k;k=f[k][2])
	{
		if (bz[f[k][1]]||f[k][1]==fa) continue;
		getlen(f[k][1],t,lenn+f[k][3]);
	}
}
int work(int t,int lenn)
{
	len[0]=0;getlen(t,0,lenn);
	stable_sort(len+1,len+len[0]+1);
	int l=1,r=len[0],sum=0;
	while (l<r)
	{
		while (len[l]+len[r]<=maxlen&&l<r) sum+=r-l,l++;
		while (len[l]+len[r]>maxlen&&l<r) r--;
	}
	return sum;
}
void find_centroid(int t,int fa)
{
	size[t]=1,son[t]=0;
	for (int k=q[t];k;k=f[k][2])
	{
		if (bz[f[k][1]]||f[k][1]==fa) continue;
		find_centroid(f[k][1],t);
		size[t]+=size[f[k][1]];
		if (size[f[k][1]]>son[t]) son[t]=size[f[k][1]];
	}
	son[t]=max(son[t],tot-size[t]);
	if (son[t]<son[cent]||!cent) cent=t;
}
void dg(int t)
{
	bz[t]=1;ans+=work(t,0);
	for (int k=q[t];k;k=f[k][2])
	{
		if (bz[f[k][1]]) continue;
		ans-=work(f[k][1],f[k][3]);
		tot=size[f[k][1]],cent=0,find_centroid(f[k][1],0);
		dg(cent);
	}
}
int main()
{
	freopen("distance.in","r",stdin);
	freopen("distance.out","w",stdout);
	scanf("%d%d",&n,&maxlen);
	for (i=1;i<=n-1;i++) scanf("%d%d%d",&o,&p,&l),insert(o,p,l),insert(p,o,l);	
	tot=n,cent=0;find_centroid(1,0);
	dg(cent);
	printf("%d\n",ans);
	fclose(stdin);fclose(stdout);
	return 0;
}