POJ 2631 Roads in the North解題報告(木の直径)

1412 ワード

転送ゲート:http://poj.org/problem?id=2631
この問題は木の直径を求める裸の問題です.大丈夫です.原理も簡単です.2回のBFSが出るので、注意しなければならないのはデータ範囲かもしれません.データを読み込むようにしましょう.重さもないし、凹みもないと判断しています.
上の波C中の空洞が実現しました
#include 
#include 
#include 
#include 
using namespace std;
const int MAXN = 20010;

struct node{
    int v, w, next;
}edge[MAXN];
int head[MAXN], cnt = 0, dis[MAXN];

void add(int u, int v, int w){
    edge[++cnt].v = v;
    edge[cnt].w = w;
    edge[cnt].next = head[u];
    head[u] = cnt;
}

int BFS(int x){
    memset(dis, 0x3f, sizeof(dis));
    queue q;
    dis[x] = 0;
    q.push(x);
    while(!q.empty()){
        int temp = q.front();
        q.pop();
        for(int i = head[temp]; i != 0;i = edge[i].next){
            if(dis[edge[i].v] > dis[temp] + edge[i].w){
                dis[edge[i].v] = dis[temp] + edge[i].w;
                q.push(edge[i].v);
                if(dis[edge[i].v] > dis[x])
                    x = edge[i].v;
            }
        }
    }
    return x;
}
int main()
{
    //freopen("input.txt", "r", stdin);
    int u, v, w;
    memset(head, 0, sizeof(head));
    while(scanf("%d %d %d", &u, &v, &w) != EOF){
        add(u, v, w);
        add(v, u, w);
    }
    int z = BFS(1);
    int e = BFS(z);
    printf("%d
", dis[e]); return 0; }