ZOJ 3820:Building Fire Stations(树的直径 Grade C)

题意：

n个点的树，边长全为1，求找出两个点，使得树上离这两个点距离最远的那个点，到这两个点（中某个点就行）的距离最小。

思路：

求树直径，找中点，删除中间那条边（如果直径上点数为奇数，则删任何一侧都可），分成两个子树，再求中心，即为答案。

代码：

//14:12
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

#define N 200100

struct Graph{
    struct Edge{
        int to;
        int next;
    }e[N*];
    int head[N];
    int p;
    void clear() {
        p = ;
        memset(head, -, sizeof(head));
    }
    void addedge(int u, int v) {
        e[p].to = v;
        e[p].next = head[u];
        head[u] = p++;
    }
}g;

int n;
int dis[N];
int que[N*];
bool vis[N];
int fa[N];

void generateDis(const Graph &g, int s) {
    memset(dis, 0x3f, sizeof(dis));
    memset(vis, , sizeof(vis));
    memset(fa, -, sizeof(fa));
    dis[s] = ;
    int hd = ;
    int tl = ;
    que[tl++] = s;
    vis[s] = ;
    while (hd < tl) {
        int now = que[hd++];
        vis[now] = ;
        for (int i = g.head[now]; i != -; i = g.e[i].next) {
            int to = g.e[i].to;
            if (dis[to] > dis[now]+) {
                dis[to] = dis[now]+;
                if (!vis[to]) {
                    vis[to] = ;
                    fa[to] = now;
                    que[tl++] = to;
                }
            }
        }
    }
}

pair<int, int> getTreeMidPoint(const Graph &g, int s, int &OUT_dis) {
    generateDis(g, s);
    for (int i = ; i <= n; i++) {
        if (dis[i] == 0x3f3f3f3f) dis[i] = -;
    }
    int u = max_element(&dis[], &dis[]+n) - dis;

    generateDis(g, u);
    for (int i = ; i <= n; i++) {
        if (dis[i] == 0x3f3f3f3f) dis[i] = -;
    }
    int v = max_element(&dis[], &dis[]+n) - dis;

    //printf("u = %d, v = %d, dis[v] = %d\n", u, v, dis[v]);

    int now = v;
    while (dis[now] > (dis[v]+)/) {
        now = fa[now];
    }

    OUT_dis = dis[v]/ + dis[v]%;
    return {fa[now], now};
}

int main() {
    int t;
    scanf("%d", &t);
    while (t--) {
        scanf("%d", &n);

        g.clear();
        for (int i = ; i < n-; i++) {
            int a, b;
            scanf("%d%d", &a, &b);
            g.addedge(a,b);
            g.addedge(b,a);
        }

        int tmpdis[];
        pair<int ,int> point = getTreeMidPoint(g, , tmpdis[]);

        //printf("all tree get (%d %d, dis=%d)\n", point.first, point.second, tmpdis[0]);

        for (int i = g.head[point.first]; i != -; i=g.e[i].next) {
            if (g.e[i].to == point.second) g.e[i].to = point.first;
        }

        for (int i = g.head[point.second]; i != -; i=g.e[i].next) {
            if (g.e[i].to == point.first) g.e[i].to = point.second;
        }

        pair<int ,int> pa = getTreeMidPoint(g, point.first, tmpdis[]);
        pair<int ,int> pb = getTreeMidPoint(g, point.second, tmpdis[]);

        printf("%d %d %d\n", max(tmpdis[], tmpdis[]), pa.second, pb.second);
    }
    return ;
}