bzoj1468 Tree

1468: Tree

Time Limit: 10 Sec   Memory Limit: 64 MB
Submit: 1038   Solved: 544
[ Submit][ Status][ Discuss]

Description

给你一棵TREE,以及这棵树上边的距离.问有多少对点它们两者间的距离小于等于K

Input

N（n<=40000） 接下来n-1行边描述管道，按照题目中写的输入 接下来是k

Output

一行，有多少对点之间的距离小于等于k

Sample Input

7
1 6 13
6 3 9
3 5 7
4 1 3
2 4 20
4 7 2
10

Sample Output

5

HINT

Source

LTC男人八题系列

树的点分治裸题，同poj1741。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define maxn 40005
using namespace std;
int n,k,sum,cnt,tot,mn,root,ans;
int head[maxn],sz[maxn],mx[maxn],dis[maxn];
bool vst[maxn];
struct edge_type{int next,to,v;}e[maxn*2];
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
	while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
inline void add_edge(int x,int y,int z)
{
	e[++cnt]=(edge_type){head[x],y,z};head[x]=cnt;
	e[++cnt]=(edge_type){head[y],x,z};head[y]=cnt;
}
inline void getroot(int x,int fa)
{
	sz[x]=1;mx[x]=0;
	for(int i=head[x];i;i=e[i].next)
	{
		int y=e[i].to;
		if (!vst[y]&&y!=fa)
		{
			getroot(y,x);
			sz[x]+=sz[y];
			mx[x]=max(mx[x],sz[y]);
		}
	}
	mx[x]=max(mx[x],sum-sz[x]);
	if (mx[x]<mx[root]) root=x;
}
inline void dfs(int x,int d,int fa)
{
	dis[++tot]=d;
	for(int i=head[x];i;i=e[i].next)
	{
		int y=e[i].to;
		if (y!=fa&&!vst[y]) dfs(y,d+e[i].v,x);
	}
}
inline int calc(int x,int d)
{
	int ret=0;
	tot=0;
	dfs(x,d,0);
	sort(dis+1,dis+tot+1);
	int i=1,j=tot;
	while (i<j)
	{
		while (dis[i]+dis[j]>k&&i<j) j--;
		ret+=j-i;
		i++;
	}//左端点递增，右端点递减，巧妙的思路 
	return ret;
}
inline void solve(int x)
{
	ans+=calc(x,0);
	vst[x]=true;
	for(int i=head[x];i;i=e[i].next)
	{
		int y=e[i].to;
		if (!vst[y])
		{
			ans-=calc(y,e[i].v);
			root=0;sum=sz[y];
			getroot(y,0);
			solve(root);
		}
	}
}
int main()
{
	n=read();
	F(i,1,n-1)
	{
		int x=read(),y=read(),z=read();
		add_edge(x,y,z);
	}
	k=read();
	mx[0]=sum=n;getroot(1,0);
	solve(root);
	printf("%d\n",ans);
	return 0;
}