poj 1741 简单树分治

原创 2016年06月01日 23:36:28

Tree
Time Limit: 1000MS   Memory Limit: 30000K
Total Submissions: 17135   Accepted: 5586

Description

Give a tree with n vertices,each edge has a length(positive integer less than 1001). 
Define dist(u,v)=The min distance between node u and v. 
Give an integer k,for every pair (u,v) of vertices is called valid if and only if dist(u,v) not exceed k. 
Write a program that will count how many pairs which are valid for a given tree. 

Input

The input contains several test cases. The first line of each test case contains two integers n, k. (n<=10000) The following n-1 lines each contains three integers u,v,l, which means there is an edge between node u and v of length l. 
The last test case is followed by two zeros. 

Output

For each test case output the answer on a single line.

Sample Input

5 4
1 2 3
1 3 1
1 4 2
3 5 1
0 0

Sample Output

8

题意是给定一棵树问你树里有多少条简单路径满足路径长度小于等于k。

假设我现在选定了一个点x,要求所有经过x的且长度小于等于k的路径数量是很好求的,把每个点到x的距离求出来,排序一下,然后双指针搞一搞就好了。那么对于不经过x的路径。递归下去做就好了。为了减少递归层数。把重心当做x。那么递归层数就只有logn层了,最终复杂度是nlog^2n

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
struct EDGE{
	int to,next;
	__int64 val;
	EDGE(){}
	EDGE(const int &to,const int &next,const __int64 &val):to(to),next(next),val(val){}
}e[202020];
int head[101010],tot;
int n;
int k;
int s[101010];
bool mark[101010];
int find_root(int u,int fa,int mxsize)
{
	s[u]=1;
	int maxn=0;
	for(int i=head[u];i>=0;i=e[i].next)
	{
		if(e[i].to==fa||mark[e[i].to])continue;
		int may=find_root(e[i].to,u,mxsize);
		if(may!=-1)return may;
		s[u]+=s[e[i].to];
		maxn=max(maxn,s[e[i].to]);
	}
	if(maxn<mxsize-s[u])maxn=mxsize-s[u];
	if(maxn*2<=mxsize)return u; 
	return -1;
}
int all[101010];
int ts[101010],tp;
void get_depth(int u,int fa,__int64 sum)
{
	ts[tp++]=sum;
	s[u]=1;
	for(int i=head[u];i>=0;i=e[i].next)
	{
		if(e[i].to==fa||mark[e[i].to])continue;
		get_depth(e[i].to,u,sum+e[i].val);
		s[u]+=s[e[i].to];
	}
}
__int64 solve(int rt)
{
	mark[rt]=true;
	__int64 ret=0;
	tot=0;
	for(int i=head[rt];i>=0;i=e[i].next)
	{
		int v=e[i].to;
		if(!mark[v]){
			tp=0;
			get_depth(v,v,e[i].val);
			sort(ts,ts+tp);
			int t=0,w=tp-1;
			while(t<w)
			{
				if(ts[w]+ts[t]>k)
				{
					w--;
				}
				else {
					ret-=(w-t);
					t++;
				}
			}
			for(int j=0;j<tp;j++){
				all[tot]=ts[j];
				if(all[tot]<=k)ret++;
				tot++;
			}
		}
	}
	int t=0,w=tot-1;
	sort(all,all+tot);
	while(t<w)
	{
		if(all[w]+all[t]>k)
		{
			w--;
		}
		else {
			ret+=(w-t);
			t++;
		}
	}
	for(int i=head[rt];i>=0;i=e[i].next)
	{
		int v=e[i].to;
		if(!mark[v]){
			int rt=find_root(v,v,s[v]);
			ret+=solve(rt);
		}
	}
	return ret;
}
int main()
{
	while(~scanf("%d%I64d",&n,&k))
	{
		if(n==0)break;
		tot=0;
		for(int i=1;i<=n;i++){head[i]=-1;mark[i]=false;}
		for(int i=1;i<n;i++)
		{
			int u,v;
			__int64 w;
			scanf("%d%d%I64d",&u,&v,&w);
			e[tot]=EDGE(v,head[u],w);
			head[u]=tot++;
			e[tot]=EDGE(u,head[v],w);
			head[v]=tot++;
		}
		int rt;
		rt=find_root(1,-1,n);
		__int64 ans=solve(rt);
		printf("%I64d\n",ans);
	}
}


版权声明:本文为博主原创文章,未经博主允许不得转载。

相关文章推荐

【树分治】poj1741 Tree

题目点这里  感觉分治没有剖分好写。。orz有个地方绕了好久。。。 漆子超的论文里解法写的挺详细的: 记Depth(i)表示点i到根结点的路径长度,Belong(i)  X (...

POJ 1741 (树的点分治入门题)

题意 : 给你一颗无根树 ,

POJ 1741 Tree DP+树的点分治

POJ 1741 Tree DP+树的点分治

POJ1741——Tree(树上分治,树的重心)

Tree Time Limit: 1000MS   Memory Limit: 30000K Total Submissions: 17804   Accepted...

poj 1741 Tree (树的分治)

TreeTime Limit: 1000MS Memory Limit: 30000K Total Submissions: 20776 Accepted: 6803Desc...

树的分治 poj 1741

求树的重心结点:用标号定义树的重心,对于一个结点K,如果把K从树中删除(连同与它相连的边一起),剩下的被分成了很多块,每一块显然又是一棵树(即剩下的部份构成了一个森林)。则给结点K所标的号就是森林中结...

poj 1741 (树的分治)

题意:给定一棵N(1 思路:看了论文《分治算法在树的路径问题中的应用》,里面讲解的很清楚,一条路径要么过根节点,要么在一颗子树中,所以用分治算法。找到树的重心作为根节点,这样每次树的节点数至少减少一...

POJ 1741 树的分治

题意就是求树上距离小于等于K的点对有多少个 n2的算法肯定不行,因为1W个点 这就需要分治。可以看09年漆子超的论文 本题用到的是关于点的分治。 一个重要的问题是,为了防止退化,所以每次都要找...

POJ-1741 Tree(树分治)

Tree Time Limit: 1000MS   Memory Limit: 30000K Total Submissions: 22767   Accepted: 7527 ...

【楼天城男人八题】【树分治|Treap+启发式合并】POJ1741 Tree

题面在这里待我先膜拜一下楼教主……首先这题是很明显的树分治 想说点什么却发现已经没什么好说了然后我们来看另一种解法:平衡树乱搞 这里用的是Treap实现对于每个节点,用Treap记录该子树每个节点...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)