2014-2015icpc 牡丹江B.Building Fire Stations (二分+树的直径)
题目链接: B. Building Fire Stations
题目大意:
给一棵n个节点的树, 要求在树上找两个点作为消防站,使得任意一个点到近的消防站的距离的最大值尽可能小, 输出最大的距离, 和两个作为消防站的标号(严格两个不同的加油站)。
数据范围: 2≤n≤2e5
解题思路:
最初是有一个二分答案的思路的, 但是并没有想好怎么贪心的去放消防站的位置, 后来队友提出了在树的直径上贪心的去放, 即在树直径的两端距离为当前二分的mid那么长的位置放两个点, 对于中间的那些点,提前处理出每个点分枝最长, 每次都去O(n) check是否合法就可以了。 之后 想了一下这么贪心很对。 之前写数的直径都是dfs直接求长度, 还没写过把直径抠出来。。 还有就是判断的时候距离得好好算一下。 别的就没啥了。 通过这道题 学会了把树的直径摘出来。
AC代码:
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <cmath>
#include <stack>
#include <cstdlib>
#include <ctime>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef double DB;
typedef pair<int, int> PII;
const int MaxN = 2e5;
vector<int>edge[MaxN + 5];
pair<int, int> q[4 * MaxN + 5];
int fa[MaxN + 5];
int seq[MaxN + 5], dis[MaxN + 5];
bool in_seq[MaxN + 5];
int n, T, tot;
int Bfs1(){
int head = 1, tail = 0;
q[++tail] = make_pair(1, 0);
while(head <= tail){
PII now = q[head];
int u = now.first, father = now.second;
for(int i = 0; i < edge[u].size(); i++){
int v = edge[u][i];
if(v != father) q[++tail] = make_pair(v, u);
}
head++;
}
return q[tail].first;
}
//这个DFS就是把直径每个点分枝最大长度求出来。
void dfs(int u, int fa){
dis[u] = 0;
for(int i = 0; i < edge[u].size(); i++){
int v = edge[u][i];
if(!in_seq[v] && v != fa) {
dfs(v, u);
dis[u] = max(dis[u], dis[v] + 1);
}
}
}
void Bfs2(int fi){
int head = 1, tail = 0;
fa[fi] = 0;
q[++tail] = make_pair(fi, 0);
while(head <= tail){
PII now = q[head];
int u = now.first, father = now.second;
for(int i = 0; i < edge[u].size(); i++){
int v = edge[u][i];
if(v != father){
q[++tail] = make_pair(v, u);
fa[v] = u;
}
}
head++;
}
int now = q[tail].first;
while(now != 0){
seq[++seq[0]] = now;
in_seq[now] = true;
now = fa[now];
}
for(int i = 1; i <= seq[0]; i++) dfs(seq[i], 0);
/* for(int i = 1; i <= seq[0]; i++) printf("%d ", seq[i]);
printf("\n");
for(int i = 1; i <= seq[0]; i++) printf("%d ", dis[seq[i]]);
printf("\n"); */
}
//用了两次BFS 求出了树的直径
bool check(int x){
int L = x + 1, R = seq[0] - x; // 两个消防站位置
for(int i = x + 2; i <= seq[0] - x - 1; i++){
int len = dis[seq[i]] + min(i - L, R - i);
if(len > x) return false;
}
return true;
}
void Clear(){
memset(edge, 0, sizeof(edge));
memset(seq, 0, sizeof(seq));
memset(dis, 0, sizeof(dis));
memset(in_seq, 0, sizeof(in_seq));
memset(fa, 0, sizeof(fa));
}
int main()
{
scanf("%d", &T);
while(T--){
scanf("%d", &n);
for(int i = 1; i < n; i++){
int u, v;
scanf("%d %d", &u, &v);
edge[u].push_back(v);
edge[v].push_back(u);
}
Bfs2(Bfs1());
int l = 0, r = seq[0], mid, ans = 0;
while(l <= r){
mid = (l + r) / 2;
if(check(mid)) ans = mid, r = mid - 1;
else l = mid + 1;
}
printf("%d ", ans);
int p1 = seq[ans + 1], p2 = seq[seq[0] - ans];
//特殊处理两个点相同的情况
if(p1 != p2) printf("%d %d\n", min(p1, p2), max(p1, p2));
else {
if(p1 + 1 <= n) p2 = p1 + 1;
else p2 = p1 - 1;
printf("%d %d\n", min(p1, p2), max(p1, p2));
}
Clear();
}
return 0;
}