# poj 1741 简单树分治

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

#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 n;
int k;
int s[101010];
bool mark[101010];
int find_root(int u,int fa,int mxsize)
{
s[u]=1;
int maxn=0;
{
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;
{
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;
{
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++;
}
}
{
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++)
{
int u,v;
__int64 w;
scanf("%d%d%I64d",&u,&v,&w);
}
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
• 2016年08月24日 16:55
• 565

## [poj1741]tree 解题报告

• TA201314
• 2016年04月21日 17:31
• 618

## [POJ 1741]Tree[树分治]

• CatGlory
• 2016年03月02日 10:07
• 245

## POJ 1741（树分治）

• u013007900
• 2015年05月27日 22:16
• 398

## poj 1741 Tree(树分治)

• gyhguoge01234
• 2017年08月29日 17:55
• 181

## poj 1741 Tree 树分治

• u014076176
• 2014年08月28日 11:42
• 354

## poj 1741 树分治入门

• HTT_H
• 2015年08月03日 23:20
• 332

## POJ 1741 树分治

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

## 【POJ 1741】Tree(树分治)

• u011332631
• 2015年06月30日 16:44
• 247

## POJ 1741 Tree【树分治】

• Yukizzz
• 2016年04月27日 19:11
• 529

举报原因： 您举报文章：poj 1741 简单树分治 色情 政治 抄袭 广告 招聘 骂人 其他 (最多只允许输入30个字)