POJ 1849 Two

求树的直径要先bfs找到任意起点能走到的最远的点,然后再以找到的点为起点bfs。
本题要求两点遍历树,实际是求两倍权和-树的直径,证明不缀。

//注意拓展点的循环的条件是i != -1还是i != 0. 
#include<cstdio>
#include<queue>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn = 100010;
int n, s, tot, end;
int sum;
int st[maxn], d[maxn], vis[maxn];

struct node{
    int to, w, next;
} edge[2*maxn];

void in(int u, int v, int w){
    edge[++tot].to = v;
    edge[tot].w = w;
    edge[tot].next = st[u];
    st[u] = tot;
}

int bfs(int start){
    memset(vis, 0, sizeof(vis));
    d[start] = 0;
    int diameter = 0;
    queue<int> q;
    q.push(start);
    vis[start] = 1;
    while(!q.empty()){
        int u = q.front();
        q.pop();
        for(int i = st[u]; i != 0; i = edge[i].next){
            int v = edge[i].to;
            if(!vis[v]){
                d[v] = d[u] + edge[i].w;
                if(d[v] > diameter){
                    diameter = d[v];
                    end = v;
                }
                vis[v] = 1;
                q.push(v);
            }
        }
    }
    return diameter;
}

int main(){
    cin >> n >> s;
    for(int i = 1, u, v, w; i <= n-1; i++){
        cin >> u >> v >> w;
        sum += w*2;
        in(u, v, w);
        in(v, u, w);
    }
    end = s;
    bfs(s);
    int dia = bfs(end);
    cout << sum - dia << endl;
    return 0;
} 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值