Roads in the North

給你一棵樹,其中邊都帶有權重,請求出此樹上的最長鍊有多長。

題解

經典Tree DP。

對於每一個點$u$,我們把他所有的小孩的鍊長倆倆串接看看能否造出更大的數值。之後,回傳過$u$點的最長鍊長度回去!

參考大大的題解

AC Code

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#include <bits/stdc++.h>

using namespace std;

typedef pair<int, int> ii;

bool dirty;
vector<ii> g[11111];

int dfs(int u, int p, int &ans)
{
    int mx = 0;
    for (auto v : g[u]) {
        if (v.first == p)
            continue;
        int cur = dfs(v.first, u, ans) + v.second;
        ans = max(ans, cur + mx); // 兩兩對接
        mx = max(mx, cur); // 要回傳的過u最長鍊
    }

    return mx;
}

void solve()
{
    int ans = 0;
    dfs(0, -1, ans);

    for (int i = 0; i < 11111; i++)
        g[i].clear();

    printf("%d\n", ans);
}

int main()
{
    char inp[111];
    dirty = false;
    while (fgets(inp, 111, stdin) != NULL) {
        if (inp[0] == '\n' || inp[0] == '\r') {
            solve();
            continue;
        }

        dirty = true;
        int u, v, w;
        sscanf(inp, "%d %d %d", &u, &v, &w);
        u--;
        v--;
        g[u].push_back(ii(v, w));
        g[v].push_back(ii(u, w));
    }

    if (dirty)
        solve();

    return 0;
}