二分+树上乱搞 zoj3820 Building Fire Stations

传送门：点击打开链接

题意：给你N个点，有N-1条边，构成了一棵树，每条边的距离都为1。现在要在N个点中选出2个点放灭火装置，要求N个点到最近的灭火装置的距离的最大值最小。求这两个点的位置。

思路：首先，一般让最大值最小，都是二分的方法。所以我们主要应该讨论，假如到灭火装置的距离最大值<=m时，是否能满足题意。选1为根节点，求出树的深度。对深度最大的那个点进行贪心策略。如果要使这个点被灭火装置覆盖，最优的方法应该是顺着边向根的方向移动m条边，或者已经到了根节点1了。那么在这个位置放一个灭火装置，最远的点肯定可以被覆盖，而且此时是能覆盖的点是最多的。然后我们从这个点出发将m条边能到达的点全部标记删除。重复上述步骤，在没有删除的点中找到深度最大的点，再向上找到放灭火装置的点，再删除一次点，最后统计是否n个点都被删除了，如果都被删除，说明m是可行的，否则m就是不可行的。

因为zoj是G++所以没办法扩栈，，所以只能全部用队列写了感觉不是很方便 。。

当然这题还要注意另外一些细节，因为灭火装置不能同时被安放在同一点，所以在第一个灭火装置放下后已经能覆盖所有的点时，第二个灭火装置再随便找个点即可。

#include<map>
#include<set>
#include<cmath>
#include<stack>
#include<queue>
#include<cstdio>
#include<cctype>
#include<string>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
#define fuck printf("fuck")
#define FIN freopen("input.txt","r",stdin)
#define FOUT freopen("output.txt","w+",stdout)
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;

const int MX = 200000 + 5;
const int ME = 400000 + 5;

int Head[MX], Next[ME], rear;
struct Edge {
    int u, v;
} E[ME];
void edge_init() {
    rear = 0;
    memset(Head, -1, sizeof(Head));
}
void edge_add(int u, int v) {
    E[rear].u = u;
    E[rear].v = v;
    Next[rear] = Head[u];
    Head[u] = rear++;
}

bool vis[MX];
int ga, gb, gans;
int F[MX], dep[MX];

struct Data {
    int u, f, d;
    Data(int _u, int _f, int _d) {
        u = _u; f = _f; d = _d;
    }
};

void build() {
    queue<PII>Q;
    Q.push(PII(1, 0));
    F[1] = -1; dep[1] = 0;

    while(!Q.empty()) {
        PII f = Q.front();
        Q.pop();

        int u = f.first, d = f.second;
        for(int i = Head[u]; ~i; i = Next[i]) {
            int v = E[i].v;
            if(v == F[u]) continue;

            F[v] = u; dep[v] = d + 1;
            Q.push(PII(v, d + 1));
        }
    }
}

int deal(int u, int x) {
    int ret = 0;
    queue<Data>Q;
    Q.push(Data(u, -1, x));

    while(!Q.empty()) {
        Data t = Q.front();
        Q.pop();

        int u = t.u, f = t.f, d = t.d;
        if(!vis[u]) ret += ++vis[u];

        if(d) for(int i = Head[u]; ~i; i = Next[i]) {
                int v = E[i].v;
                if(v == f) continue;
                Q.push(Data(v, u, d - 1));
            }
    }
    return ret;
}

bool check(int n, int x) {
    memset(vis, 0, sizeof(vis));

    int u = 1, cnt = 0;
    for(int i = 2; i <= n; i++) {
        if(dep[i] > dep[u]) u = i;
    }

    for(int i = x; i >= 1 && F[u] != -1; i--, u = F[u]);
    ga = u;
    cnt += deal(u, x);
    if(cnt == n) {
        if(ga == 1) gb = 2;
        else gb = 1;
        return true;
    }

    u = -1;
    for(int i = 1; i <= n; i++) {
        if(!vis[i] && (u == -1 || dep[i] > dep[u])) u = i;
    }

    for(int i = x; i >= 1 && F[u] != -1; i--, u = F[u]);
    gb = u;
    cnt += deal(u, x);
    return cnt == n;
}

int main() {
    int T, n; //FIN;
    scanf("%d", &T);
    while(T--) {
        edge_init();

        scanf("%d", &n);
        for(int i = 1; i <= n - 1; i++) {
            int u, v;
            scanf("%d%d", &u, &v);
            edge_add(u, v);
            edge_add(v, u);
        }
        build();

        int ma, mb, mans;
        int L = 0, R = n, m;
        while(L <= R) {
            m = (L + R) >> 1;
            if(check(n, m)) {
                ma = ga;
                mb = gb;
                mans = m;
                R = m - 1;
            } else L = m + 1;
        }
        printf("%d %d %d\n", mans, ma, mb);
    }
    return 0;
}