ZOJ Problem Set - 3820 Building Fire Stations 【树的直径 + 操作 】

题目：ZOJ Problem Set - 3820 Building Fire Stations 

题意：给出n个点，n-1条边的一棵树，然后要在两个点上建立两个消防站，让所有点的到消防站最大距离的点的这个距离最小。

分析：首先先求这个树的直径，然后在树的直径的中点处把树分成两棵树，然后在把两棵树分别取中点的最大值就是ans值。

这个题目数据有点水了感觉。。。

AC代码：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <vector>
#include <cstring>
#include <queue>
using namespace std;
const int N = 204000;
int n;
vector<int> v[N];
int vis[N],father[N];
vector<int> line;
int BFS(int s,int flag)
{
    queue<int> q;
    int e=s;
    line.clear();
    memset(vis,0,sizeof(vis));
    memset(father,-1,sizeof(father));
    vis[flag]=1;
    q.push(s);
    vis[s]=1;
    in