盒子
盒子
文章目录
  1. 题目
  2. 题意
  3. 思路
  4. 代码
  5. 更新日志

HDU 5102 The K-th Distance(想法)

题目

源地址:http://acm.hdu.edu.cn/showproblem.php?pid=5067

题意

有一棵树n个节点,每条边都为1。现在把每两个点之间的距离放在一起,按照从大到小排列。问前k个的和为多少?保证k小于等于min(1e6, n*(n-1)/2)

思路

把所有边(u,v)以及(v,u)放入队列,队列每弹出一个元素(u,v),对于所有与v相邻的点w,如果w≠u,就把(v,w)入队,这样就能一个一个生成前K小的距离。 注意到每条边实际上会入队两次,只要把K翻倍且把ans除以2即可,时间复杂度为O(n+K)

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdio>
#include<vector>
#include<queue>
#define pb push_back
#define INF 1 << 30
#define fi first
#define se second
#define debug puts("=====================");
using namespace std;
typedef long long ll;
const int N = 1e5 + 100;
const int M = 1e6 + 100;
int n, k, m;
vector<int> g[N];
struct node {
int u, v, l;
}tmp;
node q[M * 2];
int main () {
int t;
scanf("%d", &t);
while(t--) {
scanf("%d%d", &n, &k);
for (int i = 1; i <= n; i++) g[i].clear();
int u, v, l;
k *= 2;
int st = 0, ed = 0;
ll ans = 0;
for (int i = 1; i < n; i++) {
scanf("%d%d", &u, &v);
g[u].pb(v);
g[v].pb(u);
if (ed < k) q[ed++] = {u, v, 1}, ans++;
if (ed < k) q[ed++] = {v, u, 1}, ans++;
}
while(ed < k) {
tmp = q[st++];
u = tmp.u, v = tmp.v, l = tmp.l;
for (int i = 0; i < g[v].size(); i++) if (g[v][i] != u && ed < k){
q[ed++] = {v, g[v][i], l + 1};
ans += l + 1;
}
}
printf("%I64d\n", ans / 2);
}
return 0;
}

更新日志

  • 14550969 2015-08-17 22:24:35 Accepted 5102 920MS 31204K 1099 B G++ SIO__Five