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);
	}
}


POJ 1741 Tree(树的点分治、树形dp、男人八题)

【题意】求树上距离小于等于K的点对有多少个 n有10000,所以必须卡到nlogn附近。变得尤为麻烦。 首先,一对符合要求的点,它们经过的路径中一定可以找到一个根节点。 每次我们都找以这个根为“...
  • AngOn823
  • AngOn823
  • 2016年08月24日 16:55
  • 565

[poj1741]tree 解题报告

经典的点分做法就不说了(然而我写点分t了。。)线段树/平衡树启发式合并的话,就是维护子树每个节点到子树跟的距离,打一个整棵子树的标记,然后按dfs/bfs序启发式合并,合并之前先查询一下答案即可。 ...
  • TA201314
  • TA201314
  • 2016年04月21日 17:31
  • 618

[POJ 1741]Tree[树分治]

题目链接:[POJ 1741]Tree[树分治] 题意分析: 统计树上一共有多少对结点,两者之间的最短距离不超过k。 解题思路: 1e4个点,暴搜肯定是不行的。这里我们采用树分治来解决这个问题...
  • CatGlory
  • CatGlory
  • 2016年03月02日 10:07
  • 245

POJ 1741(树分治)

论文题: 树的点分治 // whn6325689 // Mr.Phoebe // http://blog.csdn.net/u013007900 #include #include...
  • u013007900
  • u013007900
  • 2015年05月27日 22:16
  • 398

poj 1741 Tree(树分治)

点分治。 如果不了解树分治的话,可以先看看09年国家集训队的论文 https://wenku.baidu.com/view/1bc2e4ea172ded630b1cb602.html https...
  • gyhguoge01234
  • gyhguoge01234
  • 2017年08月29日 17:55
  • 181

poj 1741 Tree 树分治

题目:点击打开链接
  • u014076176
  • u014076176
  • 2014年08月28日 11:42
  • 354

poj 1741 树分治入门

题意:统计距离 这道题用的树的点分治方法。点分治基于树的重心。 树的重心,定义是删除某个点后得到的最大(节点数)子树的节点数最小。性质是,可以证明删除掉重心后,每个子树的大小...
  • HTT_H
  • HTT_H
  • 2015年08月03日 23:20
  • 332

POJ 1741 树分治

【题意】求树上距离小于等于K的点对有多少个? 【解题方法】不愧是男人8题,从TLE写带WA,最后过了,经历了10+次。 一个重要的问题是,为了防止退化,所以每次都要找到树的重心然后分治下去,所谓重...
  • just_sort
  • just_sort
  • 2016年08月17日 19:42
  • 244

【POJ 1741】Tree(树分治)

之前一段时间做的,但是因为懒所以没有写题解。有点忘了现在重新做一下,还是乖乖写题解吧~ 树分治主要的在于要找重心。对于那些需要统计满足条件的路径条数的题,我们就可以想到树分治。 对于某一个点,我们可以...
  • u011332631
  • u011332631
  • 2015年06月30日 16:44
  • 247

POJ 1741 Tree【树分治】

第一次接触树分治,看了论文又照挑战上抄的代码,也就理解到这个层次了。。 以后做题中再慢慢体会学习。题目链接:http://poj.org/problem?id=1741题意:给定树和树边的权重,求有...
  • Yukizzz
  • Yukizzz
  • 2016年04月27日 19:11
  • 529
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:poj 1741 简单树分治
举报原因:
原因补充:

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