SPOJ 9126 Time to live

点击打开SPOJ 9126

题意：给定一个n台计算机的网络的连接图，这个图是一棵树的形式。现在要以某一台计算机为路由器，问其它的计算机到路由器的最长的距离的最小值

思路：给定一个树，我们能够求出树的直径。那么直径的两端的距离是最长的，那么路由器的选择肯定是在树的直径上面的某一点，因为要距离最小因此选择中间的点肯定能够满足。那么maxLen为直径的话，ans为(maxLen+1)/2

代码：

#include<vector>
#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;

const int MAXN = 100010;

struct Node{
    int num;
    int step;
};

int n , start , maxLen;
bool vis[MAXN];
vector<int>v[MAXN]; 
queue<Node>q;

void init(){
    for(int i = 0 ; i < n ; i++)
        v[i].clear();
    int x , y;
    for(int i = 1 ; i < n ; i++){
        scanf("%d%d" , &x , &y);
        v[x].push_back(y);
        v[y].push_back(x);
    }
}

void bfs(){
    maxLen = 0;
    while(!q.empty()){
        Node tmp = q.front();
        q.pop();
        int x = tmp.num;
        int size = v[x].size();
        bool isOk = false;
        vis[x] = true;
        for(int i = 0 ; i < size ; i++){
            if(!vis[v[x][i]]){
               q.push((Node){v[x][i] , tmp.step+1}); 
               isOk = true;
            }       
        }
        if(!isOk && maxLen < tmp.step){
           maxLen = tmp.step;
           start = x; 
        }
    }
}

int solve(){
    while(!q.empty())
        q.pop();
    q.push((Node){0,0});
    memset(vis , false , sizeof(vis));
    bfs();

    while(!q.empty())
        q.pop();
    q.push((Node){start,0});
    memset(vis , false , sizeof(vis));
    bfs();
    
    return (maxLen+1)/2;
}

int main(){
    int cas;
    scanf("%d" , &cas);
    while(cas--){
        scanf("%d" , &n);
        init();
        printf("%d\n" , solve());
    }
    return 0;
}