给你1e5个节点的树,(⊙﹏⊙)

你能求出又几对节点的距离小于k吗??(分治NB!)

这只是一个板子题,树上分治没有简单题呀!(一个大佬说的)

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#define maxn 10020
using namespace std;
struct Node {
int p;
int val;
Node(int _p, int _val) :p(_p), val(_val) {}
};
vector<Node>G[maxn];
void insert(int be, int en, int len) {
G[be].push_back(Node(en, len));
}
int dis[maxn];//距离
int son[maxn];
int ans[maxn];
int vis[maxn];
int d[maxn];
int chal;
int k;
int n;
int cntv = 0;
int root;
int ln = 0; void find_root(int x, int fa) {//找重心
son[x] = 1;
ans[x] = 0;
for (int i = 0; i < G[x].size(); i++) {
int p = G[x][i].p;
if (p == fa || vis[p]) continue;
find_root(p, x);
son[x] += son[p];
ans[x] = max(son[p], ans[x]);
}
ans[x] = max(cntv - son[x], ans[x]);//cntv当前树重量 if (ans[x] < ans[root]) root = x; //尽量小
} void get_dis(int x, int fa) {//计算各个点到根的距离
dis[ln++] = d[x];
for (int i = 0; i < G[x].size(); i++) {
int p = G[x][i].p;
if (fa == p || vis[p]) continue;
d[p] = d[x] + G[x][i].val;
get_dis(p, x);
} }
int cal(int x, int val) {//val要是零,x就是根,否则就是子树
d[x] = val;
ln = 0;
get_dis(x, 0);
sort(dis, dis + ln);
int l = 0, r = ln - 1, res = 0;
while (l < r) {
if (dis[l] + dis[r] <= k) {
res += r - l;
l++;
}
else r--;
}
return res;
}
int slove(int x) {
chal += cal(x, 0);
vis[x] = 1;
for (int i = 0; i < G[x].size(); i++) {
int p = G[x][i].p;
if (vis[p]) continue;
chal -= cal(p, G[x][i].val);
cntv = son[p];
root = 0;
ans[0] = 0x3f3f3f3f;
find_root(p, 0);
slove(root);
}
return 0;
}
int main() {
int be, en, len;
while (~scanf("%d%d", &n, &k)) {
if (n == 0 && k == 0) break;
memset(vis, 0, sizeof(vis));
ans[0] = 0x3f3f3f3f;
// memset()
for (int i = 0; i <= n; i++) G[i].clear();
for (int i = 0; i < n - 1; i++) {
scanf("%d %d %d", &be, &en, &len);
insert(be, en, len);
insert(en, be, len);
}
root = 0;
cntv = n;
find_root(1, 0);
chal = 0;
slove(root);
printf("%d\n", chal);
}
return 0;
}

  

最新文章

  1. GitHub实战系列汇总篇
  2. ES6 对象增强和结构赋值
  3. MySql之安装配置
  4. 人人都是 DBA(XV)锁信息收集脚本汇编
  5. 理解 QEMU/KVM 和 Ceph(1):QEMU-KVM 和 Ceph RBD 的 缓存机制总结
  6. 39. Recover Binary Search Tree &amp;&amp; Validate Binary Search Tree
  7. 水溶彩铅的特点&amp;技法运用
  8. 安装grunt
  9. 批处理中的echo命令图文详解
  10. Shell 操作练习2
  11. [IT学习]微软如何做网站内容治理
  12. Contributing to Open Source on GitHub(转)
  13. Jmeter学习笔记
  14. iOS Learning
  15. Android软件设置自动检查更新
  16. hdoj:2034
  17. Python开发【第三篇】:Python函数
  18. Vsphere初试——使用Vsphere client
  19. JSP之Cookie对象使用
  20. MySQL left join right join inner join

热门文章

  1. 【git基本操作】总结
  2. 一维数组的初始化及遍历 Day06
  3. 一个 PHP 面试题
  4. OpenResty,X-WAF防火墙相关
  5. laravel 定时任务通过队列发送邮件
  6. POJ2976 题解 0/1分数规划入门题 二分
  7. 【codeforces 777E】Hanoi Factory
  8. springboot 实现 aop
  9. java 文件读写工具 FileUtil
  10. H3C PPP会话流程