HDU 3848 CC On The Tree 树形DP

题意:

给出一棵边带权的树,求距离最近的一对叶子。

分析:

通过DFS计算出\(min(u)\):以\(u\)为根的子树中最近叶子到\(u\)的距离。
然后维护一个前面子树\(v_i\)中叶子到\(u\)距离的最小值,就可以用这个最小值+当前子树中叶子到\(u\)的最短距离来更新答案。
如果根节点也是叶子节点的话,再用\(min(root)\)更新一下答案。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int maxn = 10000 + 10;
const int INF = 0x3f3f3f3f;

struct Edge
{
    int v, w, nxt;
    Edge() {}
    Edge(int v, int w, int nxt): v(v), w(w), nxt(nxt) {}
};

int ecnt, head[maxn];
Edge edges[maxn * 2];

void AddEdge(int u, int v, int w) {
    edges[ecnt] = Edge(v, w, head[u]); head[u] = ecnt++;
    edges[ecnt] = Edge(u, w, head[v]); head[v] = ecnt++;
}

int n;

int ans;
int minv[maxn];
int child[maxn];

void dfs(int u, int p) {
    minv[u] = INF;
    int minp = INF;
    child[u] = 0;
    for(int i = head[u]; ~i; i = edges[i].nxt) {
        int v = edges[i].v;
        if(v == p) continue;
        child[u]++;
        int w = edges[i].w;
        dfs(v, u);
        minv[u] = min(minv[u], minv[v] + w);
        ans = min(ans, minp + minv[v] + w);
        minp = min(minp, minv[v] + w);
    }
    if(!child[u]) minv[u] = 0;
}

int main()
{
    while(scanf("%d", &n) == 1 && n) {
        ecnt = 0;
        memset(head, -1, sizeof(head));
        for(int i = 1; i < n; i++) {
            int u, v, w; scanf("%d%d%d", &u, &v, &w);
            AddEdge(u, v, w);
        }

        ans = INF;
        dfs(1, 0);
        if(child[1] == 1) ans = min(ans, minv[1]);
        printf("%d\n", ans);
    }

    return 0;
}
posted @ 2016-04-05 19:15  AOQNRMGYXLMV  阅读(...)  评论(... 编辑 收藏