【POJ1741】Tree（点分治）

【POJ1741】Tree（点分治）

题面

Vjudge

题目大意：

求树中距离小于\(K\)的点对的数量

题解

完全不觉得点分治了。。

简直\(GG\)，更别说动态点分治了。。。

于是来复习一下。

对于每一层分治重心

求出它到子树中任意点的距离

然后\(two-pointers\)计算满足大于\(K\)的点对的数目，累加到答案中，

但是同一棵子树内的会算重

所以再对于每一棵子树，减去字数内两两满足要求的点对

完全不记得了

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 11111
inline int read()
{
    RG int x=0,t=1;RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
struct Line{int v,next,w;}e[MAX<<1];
int h[MAX],cnt=1;
inline void Add(int u,int v,int w){e[cnt]=(Line){v,h[u],w};h[u]=cnt++;}
int root,Size,rmx;
int S[MAX],dep[MAX],tot;
int size[MAX],K,n,ans;
bool vis[MAX];
void Getroot(int u,int ff)
{
	size[u]=1;int mx=0;
	for(int i=h[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(v==ff||vis[v])continue;
		Getroot(v,u);
		size[u]+=size[v];mx=max(mx,size[v]);
	}
	mx=max(mx,Size-size[u]);
	if(mx<rmx)rmx=mx,root=u;
}
void getdep(int x,int ff)
{
	S[++tot]=dep[x];
	for(int i=h[x];i;i=e[i].next)
	{
		int v=e[i].v;
		if(v==ff||vis[v])continue;
		dep[v]=dep[x]+e[i].w;
		getdep(v,x);
	}
}
int Calc(int u,int pre)
{
	tot=0;dep[u]=pre;
	getdep(u,0);
	int l=1,r=tot,ret=0;
	sort(&S[1],&S[tot+1]);
	while(l<r)
	{
		if(S[l]+S[r]<=K)ret+=r-l,++l;
		else --r;
	}
	return ret;
}
void Solve(int u)
{
	ans+=Calc(u,0);vis[u]=true;
	for(int i=h[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(vis[v])continue;
		ans-=Calc(v,e[i].w);
		Size=rmx=size[v];
		Getroot(v,u);
		Solve(root);
	}
}
int main()
{
	while(233)
	{
		rmx=Size=n=read();K=read();
		if(!n&&!K)break;
		memset(vis,0,sizeof(vis));ans=0;
		memset(h,0,sizeof(h));cnt=1;
		for(int i=1;i<n;++i)
		{
			int u=read(),v=read(),w=read();
			Add(u,v,w);Add(v,u,w);
		}
		Getroot(1,0);
		Solve(root);
		printf("%d\n",ans);
	}
}