# poj 1741 简单树分治

120人阅读 评论(0)

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

0
0

【直播】机器学习&数据挖掘7周实训--韦玮
【套餐】系统集成项目管理工程师顺利通关--徐朋
【直播】3小时掌握Docker最佳实战-徐西宁
【套餐】机器学习系列套餐（算法+实战）--唐宇迪
【直播】计算机视觉原理及实战--屈教授
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之矩阵--黄博士
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之凸优化--马博士
【套餐】Javascript 设计模式实战--曾亮

* 以上用户言论只代表其个人观点，不代表CSDN网站的观点或立场
个人资料
• 访问：4622次
• 积分：190
• 等级：
• 排名：千里之外
• 原创：15篇
• 转载：0篇
• 译文：0篇
• 评论：0条