POJ 1985 Cow Marathon 动态规划/深搜

POJ 1985 Cow Marathon 动态规划/深搜

思路：
1985也可以用1986的程序改改就行了。
但是觉得不用什么算法也是可以做出1985的。

想了一下，发现：
路径的最大值一定存在于两个叶子节点中。
如果只有一个叶子，那整个树就是一条直线了。

由于我们只是考虑叶子节点。那么对于每一个非叶子节点，我们只需要找出它下面的所有节点中，离它最远的两个叶子就行了。
这两个叶子节点的距离也就有可能成为答案。
对于每个点，我们只需要保存一个值，就是该点下面的所有节点中，距离它最远的一个叶子节点，和它的距离。
对于每个点，遍历完它的孩子之后，就知道“离它最远的两个叶子的距离”了。

注意：
代码里需要处理“一条直线连着几个点”这种情况，将这样的几个点缩成一个点比较好。不做这个处理一定会爆栈。最后一个数据是一条直线。（阴险）

这份代码跑了141MS，还算可以，呵呵。应该比直接用lca要快。

#include  < stdio.h >

#define  MAX_N 40032

struct  edge_node  {
    struct edge_node *next;
    int idx, len;
} ;
struct  edge_node edges[MAX_N * 2 ];

struct  tree_node  {
    struct edge_node *edge;
    int visited;
} ;
struct  tree_node tree[MAX_N];
int  max_val;

__inline  void  add_edge( int  idx,  int  a,  int  b,  int  len)
{
    struct edge_node *e = &edges[idx];
    e->idx = b;
    e->len = len;
    e->next = tree[a].edge;
    tree[a].edge = e;
}

int  dfs( int  idx)
{
    struct edge_node *e;
    int sum, cnt, arr[2], r;

    sum = 0;
    while (1) {
        tree[idx].visited = 1;
        cnt = 0;
        for (e = tree[idx].edge; e; e = e->next)
            cnt += !tree[e->idx].visited;
        if (!cnt)
            return sum;
        if (cnt > 1)
            break;
        for (e = tree[idx].edge; tree[e->idx].visited; e = e->next);
        sum += e->len;
        idx = e->idx;
    }

    arr[0] = arr[1] = 0;
    for (e = tree[idx].edge; e; e = e->next) {
        if (tree[e->idx].visited)
            continue;
        r = dfs(e->idx) + e->len;
        if (r >= arr[1]) {
            arr[0] = arr[1];
            arr[1] = r;
        } else if (r >= arr[0])
            arr[0] = r;
    }

    r = arr[0] + arr[1];
    if (r > max_val)
        max_val = r;

    return arr[1] + sum;
}

int  main()
{
    int m, n, a, b, len, i;
    char str[16];

    freopen("e:\\test\\in.txt", "r", stdin);

    scanf("%d%d", &n, &m);
    for (i = 0; i < m*2; i += 2) {
        scanf("%d%d%d%s", &a, &b, &len, str);
        add_edge(i, a, b, len);
        add_edge(i + 1, b, a, len);
    }

    for (i = 1; i <= n; i++) {
        if (tree[i].visited)
            continue;
        a = dfs(i);
        if (a > max_val)
            max_val = a;
    }
    printf("%d\n", max_val);

    return 0;
}