题意:给一棵树,n个节点,给定一个数k,求任意满足dist(a,b)<=k的点对的数量。

思路:

  这道题的思路比较简单,但是细节很多。

  此题可以用分治法,如何分治?

  (1)如果path(a,b)不经过根,那么肯定在根的某棵子树下,递归解决。

  (2)如果path(a,b)经过根,那么肯定在根的不同子树下,处理它。

  怎样处理?如果知道了每棵子树中的节点到根的距离,暂且将每棵子树的节点分到每个独立的桶里。每个节点都可以和其他桶里的点组成点对,只要距离<=k的话就满足要求了。逐个算可能超时了,用个简单点的方法。在这里我们不需要知道节点是谁,只需要知道距离,所以将所有节点到根的距离取出来,排个序,用两个指针在线性复杂度就可以解决本节点下的点对统计了。

  但是如何去重?因为所有子树中的节点全部都混着排序了,估计会有挺多对是在同棵子树下的节点被统计到了,这暂时不需要,这是要递归解决的,那么就要去掉这些同棵树下的点对。计算的方法都是一样的。

  但是还有一种情况让你TLE,就是单链时的情况,若每次以孩子来递归下去解决,可能就不行了。但是可以发现到,计算这棵子树时,完全也不用经过根,那么不妨直接找出这棵子树中的重心作为根来解决这个子问题。复杂度主要是在排序的地方,用了找重心的方法可以保证每个点最多被排序2*logn次,所以总复杂度为O(nlog2n)。

  如下图,红色点为重心,红色的线为虚拟边。左图是原树,右图是分治的过程。就像是每个重心隔开了一些子树一样。

  

  有论文可以看。

 //#include <bits/stdc++.h>
#include <vector>
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define pii pair<int,int>
#define INF 0x3f3f3f3f
#define LL long long
using namespace std;
const int N=;
int n, k, root, edge_cnt, ans, num, big, vis[N];
vector<int> vect[N];
struct node
{
int from,to,len;
node(){};
node(int from,int to,int len):from(from),to(to),len (len){};
}edge[N*]; void add_edge(int from,int to,int len)
{
edge[edge_cnt]=node(from,to,len);
vect[from].push_back(edge_cnt++);
} int get_size(int t,int far)
{
int cnt=;
for(int i=; i<vect[t].size(); i++)
{
node e=edge[vect[t][i]];
if(e.to!=far&&!vis[e.to]) cnt+=get_size(e.to, t);
}
return cnt;
} int get_root(int t,int far) //求重心:root。
{
int cur=, sum=;
for(int i=; i<vect[t].size(); i++)
{
node e=edge[vect[t][i]];
if( e.to!=far && !vis[e.to] )
{
int tmp=get_root(e.to, t);
sum+=tmp;
cur=max(cur, tmp);
}
}
cur=max(cur, num-sum);
if(cur<big) big=cur, root=t; //更新root
return sum;
} vector<int> seq;
void get_all(int t,int far,int len) //获取子树t的所有节点
{
seq.push_back(len);
if(len>=k) return ; //剪枝
for(int i=; i<vect[t].size(); i++)
{
node e=edge[vect[t][i]];
if( !vis[e.to] && e.to!=far ) get_all(e.to, t, len+e.len);
}
} int cal(int t,int len) //计算子树t的内部点对。
{
seq.clear();
get_all(t, , len); //将以t为根的整棵子树的len装进去seq
sort(seq.begin(), seq.end());
int tmp=, L=, R=seq.size()-;
while( L<R )
{
if( seq[L]+seq[R]<=k ) tmp+=R-L,L++;
else R--;
}
return tmp;
} void solve(int t) //主函数
{
vis[t]=;
ans+=cal(t, ); //计算子树t的答案
for(int i=; i<vect[t].size(); i++)
{
node e=edge[vect[t][i]];
if( !vis[e.to] )
{
ans-=cal(e.to, e.len); //去重
big=INF;root=-;num=get_size(e.to, -); //用于找重心
get_root(e.to, -);
solve(root);
}
}
} void init()
{
for(int i=; i<=n; i++)vect[i].clear();
memset(vis, , sizeof(vis));
ans=edge_cnt=;
} int main()
{
//freopen("input.txt", "r", stdin);
int a,b,c;
while( scanf("%d%d",&n,&k), n+k )
{
init();
for(int i=; i<n; i++)
{
scanf("%d%d%d",&a,&b,&c);
add_edge(a,b,c);
add_edge(b,a,c);
}
num=n;big=INF;root=-; //找重心用的。
get_root(,-); //找重心
solve( root );
printf("%d\n", ans);
}
return ;
}

AC代码

POJ 1741 Tree (树的分治,树的重心)的更多相关文章

  1. POJ 1741 Tree 树上点分治

    题目链接:http://poj.org/problem?id=1741 题意: 给定一棵包含$n$个点的带边权树,求距离小于等于K的点对数量 题解: 显然,枚举所有点的子树可以获得答案,但是朴素发$O ...

  2. POJ 1741 Tree (点分治)

                                                                        Tree Time Limit: 1000MS   Memory ...

  3. poj 1741 Tree(点分治)

    Tree Time Limit: 1000MS   Memory Limit: 30000K Total Submissions: 15548   Accepted: 5054 Description ...

  4. poj 1741 Tree(树的点分治)

    poj 1741 Tree(树的点分治) 给出一个n个结点的树和一个整数k,问有多少个距离不超过k的点对. 首先对于一个树中的点对,要么经过根结点,要么不经过.所以我们可以把经过根节点的符合点对统计出 ...

  5. POJ 1741.Tree and 洛谷 P4178 Tree-树分治(点分治,容斥版) +二分 模板题-区间点对最短距离<=K的点对数量

    POJ 1741. Tree Time Limit: 1000MS   Memory Limit: 30000K Total Submissions: 34141   Accepted: 11420 ...

  6. POJ 1741 Tree 求树上路径小于k的点对个数)

                                                                                                 POJ 174 ...

  7. POJ 1741 Tree (树分治入门)

    Tree Time Limit: 1000MS   Memory Limit: 30000K Total Submissions: 8554   Accepted: 2545 Description ...

  8. POJ 1741 Tree 树的分治

    原题链接:http://poj.org/problem?id=1741 题意: 给你棵树,询问有多少点对,使得这条路径上的权值和小于K 题解: 就..大约就是树的分治 代码: #include< ...

  9. POJ 1741 Tree 树形DP(分治)

    链接:id=1741">http://poj.org/problem?id=1741 题意:给出一棵树,节点数为N(N<=10000),给出N-1条边的两点和权值,给出数值k,问 ...

  10. POJ 1741 Tree(点分治点对<=k)

    Description Give a tree with n vertices,each edge has a length(positive integer less than 1001). Def ...

随机推荐

  1. html+css构成的框架,可自行改造

    运行效果 代码下载地址:http://pan.baidu.com/s/1eSeBh2E

  2. HUD-1548

    A strange lift Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)To ...

  3. sql server 2008 R2 升级与安装遇到的问题

    因工作需要,遂把以前的2008升级到r2,升级失败,具体原因忘了,卸载2008,清了注册表删了文件,结果安装的时候失败了,如下图: 下一步-有错误日志和错误的序列号,错误日志在C:\Program F ...

  4. 给source insight添加.cc的C++文件后缀识别(转载)

    转自:http://blog.chinaunix.net/uid-9950859-id-99172.html 今天在读mysql代码的时候,发现.cc结尾的文件都没有添加进来,google下了,发现原 ...

  5. Entity FrameWork 5 增删改查 & 直接调用sql语句 ?

    #region 1.0 新增 -void Add() /// <summary> /// 1.0 新增 /// </summary> static void Add() { / ...

  6. 洛谷 - P1593 - 因子和 - 费马小定理

    类似的因为模数比较小的坑还有卢卡斯定理那道,也是有时候逆元会不存在,因为整除了.使用一些其他方法避免通过逆元. https://www.luogu.org/fe/problem/P1593 有坑.一定 ...

  7. 201621123016 《Java程序设计》第九周学习总结

    1. 本周学习总结 1.1 以你喜欢的方式(思维导图或其他)归纳总结集合与泛型相关内容. 2. 书面作业 本次作业题集集合 1. List中指定元素的删除(题集题目) 1.1 实验总结.并回答:列举至 ...

  8. bzoj 2039: [2009国家集训队]employ人员雇佣【最小割】

    一开始在https://www.cnblogs.com/lokiii/p/10770919.html基础上连(i,j,b[i][j])建了个极丑的图T掉了--把dinic换成isap勉强能卡过 首先因 ...

  9. perl 处理特殊字符

    如果大家想使用perl 来处理一些特殊字符,例如"del"这种字符,就需要使用到chr() 函数了 例如大家如果向一个文本中写入以下内容 aaaa0x1270x1bccccc 这时 ...

  10. assembly x86(nasm)的日常

    cs的日常打卡. data segment ENG db 'SUNdayS Coming I Wanna Drive My Car,SUN,SUN$' ;43,35 sun1 db 'SUN' swc ...