牛客練習賽27【C 水圖 dfs求最長路】

連結：https://www.nowcoder.com/acm/contest/188/C
來源：牛客網

題目描述

小w不會離散數學，所以她van的圖論遊戲是送分的

小w有一張n個點n-1條邊的無向聯通圖,每個點編號為1~n,每條邊都有一個長度
小w現在在點x上
她想知道從點x出發經過每個點至少一次，最少需要走多少路

輸入描述:

第一行兩個整數 n,x,代表點數,和小w所處的位置
第二到第n行,每行三個整數 u,v,w,表示u和v之間有一條長為w的道路

輸出描述:

一個數表示答案

示例1

輸入

3 1
1 2 1
2 3 1

輸出

2

備註:

1 ≤ n ≤ 50000 , 1 ≤ w ≤ 2147483647

題解：發現從 x 出發,只有一條路徑可以只經過一次,其他的都會經過兩次 。找出這條最長的路徑,用所有路徑權值之和*2,減去這條路徑的就好了。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#define ll long long
#define INF 0x3f
using namespace std;
const int maxn = 50000+7;
int head[maxn], ver[maxn<<1];
ll edge[maxn<<1];
int Next[maxn];
int vis[maxn];
ll maxx;
int n, x;
int tot = 0;
ll ans = 0, sum = 0;
void add(int x, int y, ll z) {
    ver[++tot] = y, edge[tot] = z;
    Next[tot] = head[x];
    head[x] = tot;
}
void dfs(int x, int fa, ll step){
    maxx = max(step, maxx);
    for(int i = head[x]; i; i = Next[i]){
        if(ver[i] == fa) continue;
        dfs(ver[i], x, step+edge[i]);
    }
}

int main()
{
    int u, v;
    ll w;
    scanf("%d %d", &n, &x);
    for(int i = 0; i < n-1; i++){
        scanf("%d %d %lld", &u, &v, &w);
        add(u, v, w);
        add(v, u, w);
        ans += w;
    }
    dfs(x, 0, 0);
    printf("%lld\n", ans*2 - maxx);
    return 0;
}
/*
4 3
1 2 1
1 3 1
3 4 1
*/