hdu 4514 湫湫系列故事——设计风景线（树DP）

Problem - 4514

　　这是今晚的腾讯第三场初赛题中的1005。虽然我没有参加这场比赛，不过我还是尝试做了一下今晚的变态题。

　　这道题是要你判一幅图中是否有环，有环的话就输出“YES”，没有的话就输出森林里最长的一条路径。题目是中文的，就解释这么多吧！

　　做法就是直接dfs，每次保存两个值，一个是子树单链的最长长度，另一个是子树中最长的完整链的长度。

　　不过dfs有一个问题就是会爆栈，如果不想改bfs的话就要加上栈挂。

　　哈哈，最意想不到的是我居然一次做就对树DP了，除了那狗血的RE。-.-

代码如下：

#pragma comment(linker, "/STACK:102400000,102400000")

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <vector>
#include <algorithm>
#include <set>
#include <iostream>
#include <map>

using namespace std;

#define REP(i, n) for (int i = 0; i < (n); i++)
#define REP_1(i, n) for (int i = 1; i <= (n); i++)
#define PB push_back
#define MPR make_pair
#define SZ(x) ((int) (x).size())
#define FI first
#define SE second
#define _clr(x) memset(x, 0, sizeof(x))
#define ALL(x) (x).begin(), (x).end()

typedef vector<int> VI;
typedef pair<int, int> PII;
typedef vector<PII> VPII;

const int N = 1e5 + 100;

bool vis[N];
VPII rel[N];

void input(int n, int m) {
    REP_1(i, n) rel[i].clear();
    int x, y, w;
    REP(i, m) {
        scanf("%d%d%d", &x, &y, &w);
        rel[x].PB(MPR(y, w));
        rel[y].PB(MPR(x, w));
    }
}

PII dfs(int x) {
    VI dis;
    dis.clear();
    int cnt = 0, mx = 0;
    vis[x] = true;
    REP(i, SZ(rel[x])) {
        int y = rel[x][i].FI, d = rel[x][i].SE;
        if (vis[y]) {
            cnt++;
            if (cnt > 1) return MPR(-1, -1);
            continue;
        }
        PII tmp = dfs(y);
        if (~tmp.FI) {
            dis.PB(d + tmp.FI);
            mx = max(mx, tmp.SE);
        } else {
            return MPR(-1, -1);
        }
    }
    sort(ALL(dis));
    int sz = SZ(dis);
    if (sz >= 2) return MPR(dis[sz - 1], max(dis[sz - 1] + dis[sz - 2], mx));
    else if (sz == 1) return MPR(dis[sz - 1], dis[sz - 1]);
    else return MPR(0, 0);
}

int work(int n) {
    int mx = 0;
    _clr(vis);
    REP_1(i, n) {
        if (!vis[i]) {
            int tmp = dfs(i).SE;
            if (~tmp) mx = max(mx, tmp);
            else return -1;
        }
    }
    return mx;
}

int main() {
    int n, m;
    while (~scanf("%d%d", &n, &m)) {
        input(n, m);
        int ans = work(n);
        if (~ans) printf("%d\n", ans);
        else puts("YES");
    }
    return 0;
}

——written by Lyon