2014-2015icpc 牡丹江 B.Building Fire Stations (二分+树的直径)

2014-2015icpc 牡丹江B.Building Fire Stations (二分+树的直径)

---

题目链接: B. Building Fire Stations

题目大意:
给一棵n个节点的树， 要求在树上找两个点作为消防站，使得任意一个点到近的消防站的距离的最大值尽可能小， 输出最大的距离， 和两个作为消防站的标号(严格两个不同的加油站)。

数据范围: $2 \leq n \leq 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;
}