hdu 4514湫湫系列故事——设计风景线

湫湫系列故事——设计风景线

Time Limit: 6000/3000 MS (Java/Others) Memory Limit: 65535/32768 K (Java/Others)
Total Submission(s): 5769 Accepted Submission(s): 1056

Problem Description
　　随着杭州西湖的知名度的进一步提升，园林规划专家湫湫希望设计出一条新的经典观光线路，根据老板马小腾的指示，新的风景线最好能建成环形，如果没有条件建成环形，那就建的越长越好。
　　现在已经勘探确定了n个位置可以用来建设，在它们之间也勘探确定了m条可以设计的路线以及他们的长度。请问是否能够建成环形的风景线？如果不能，风景线最长能够达到多少？
　　其中，可以兴建的路线均是双向的，他们之间的长度均大于0。

Input
　　测试数据有多组，每组测试数据的第一行有两个数字n, m，其含义参见题目描述；
　　接下去m行，每行3个数字u v w，分别代表这条线路的起点，终点和长度。

　　[Technical Specification]
　　1. n<=100000
　　2. m <= 1000000
　　3. 1<= u, v <= n
　　4. w <= 1000

Output
　　对于每组测试数据，如果能够建成环形（并不需要连接上去全部的风景点），那么输出YES，否则输出最长的长度，每组数据输出一行。

Sample Input
3 3
1 2 1
2 3 1
3 1 1

Sample Output
YES

题解:判环+求树的直径，本来是用并查集判环，但内存一直超就直接dfs了，树形dp求输出直径，g[u]表示往儿子方向走的最长路，f[u]表示第一步往父亲方向走的最长路=max(f[pre[u]], g[v=pre->v]+w(pre,v)] + w(u->pre);v!=u && v != pre[pre].

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
#define INF 0x3f3f3f3f
#define N 100001
struct node {
    int v;
    int w;
    int next;
}edge[N*20];
int n, m;
//int father[N];
int head[N], cnt;
int f[N], g[N], p[N];
bool vis[N];
void addedge(int u, int v, int w)
{
    edge[++cnt].v = v;
    edge[cnt].w = w;
    edge[cnt].next = head[u];
    head[u] = cnt;
}
/*
int find_father(int x)
{
    int a = x;
    while(x != father[x])
        x = father[x];

    while(a != father[a]) {
        int z = a;
        a = father[a];
        father[z] = x;
    }
    return x;
}*/
bool dfs(int u, int pre)
{
    vis[u] = true;
    for(int i = head[u]; ~i; i =edge[i].next) {
        int v = edge[i].v;
        if(v == pre)
            continue;
        if(vis[v]||dfs(v, u))   return true;
    }
    return false;
}
void dfs1(int pos, int pre)
{
    vis[pos] = true;
    g[pos] = 0;
    for(int i = head[pos]; ~i; i = edge[i].next) {
        int v = edge[i].v;
        if(v == pre)
            continue;
        dfs1(v, p[v]=pos);
        g[pos] = max(g[pos], g[v]+edge[i].w);
    }
}
void dfs2(int pos, int pre)
{
    int t = 0;
    f[pos] = f[pre];
    for(int i = head[pre]; ~i; i = edge[i].next) {
        int v = edge[i].v;
        if(v == p[pre])
            continue;
        if(v == pos)
            t = edge[i].w;
        else
            f[pos] = max(f[pos], g[v]+edge[i].w);
    }
    f[pos] += t;
    for(int i = head[pos]; ~i; i = edge[i].next) {
        int v = edge[i].v;
        if(v != pre)
            dfs2(v, pos);
    }
}
int main(int argc, char const *argv[])
{
    while(scanf("%d%d", &n, &m) != EOF) {
        memset(head, -1, sizeof(head));
        for(int i = 0; i <= n; i++)
            vis[i] = false;
        int u, v, val;
        cnt = 0;
        bool flag = false;
        for(int i = 1; i <= m; i++) {
            scanf("%d%d%d", &u, &v, &val);

            addedge(u, v, val);
            addedge(v, u, val);
            /*
            int faA = find_father(u);
            int faB = find_father(v);
            if(faA == faB) {
                flag = true;
            } else {
                father[u] = v;
            }
            */
        }
        for(int i = 1; i <= n; i++) {
            if(!vis[i]) {
                if(dfs(i, 0)) {
                    flag = true;
                    break;
                }
            }
        }
        memset(vis, false, sizeof(vis));
        if(flag)
            printf("YES\n");
        else {
            for(int i = 1; i <= n;i ++) {
                if(!vis[i]) {
                    dfs1(i, 0);
                    dfs2(i, 0);
                }
            }
            int ans = -INF;
            for(int i = 1; i <= n; i++) {
                ans = max(ans, max(g[i], f[i]));
            }
            printf("%d\n", ans);
        }
    }
    return 0;
}