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

湫湫系列故事——设计风景线
Time Limit:3000MS Memory Limit:32768KB 64bit IO Format:%I64d & %I64u
Submit

Status

Practice

HDU 4514
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

---

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>

using namespace std;

const int vmaxn=100010;

struct Edge
{
    int to,w;
    Edge(int a,int b)
    {
        to=a;
        w=b;
    }
};

int pre[vmaxn],dp[vmaxn],ans;
vector<Edge> edge[vmaxn];



int Find(int x)
{
    int r=x;

    while (pre[r]!=r)
    {
        r=pre[r];
    }

    int i=x,j;
    while (pre[i]!=r)
    {
        j=pre[i];
        pre[i]=r;
        i=j;
    }
    return r;
}



void dfs(int u,int fa)
{
    int maxx=0;
    for (int i=0; i<edge[u].size(); i++)
    {
        int to=edge[u][i].to;
        int w=edge[u][i].w;
        if (to==fa)
            continue;
        dfs(to,u);
        ans=max(ans,dp[to]+maxx+w);
        maxx=max(maxx,dp[to]+w);
    }
    dp[u]=maxx;
}

int main()
{
    int n,m,a,b,c,i,j,tm;
    while (scanf("%d%d",&n,&m)==2)
    {
        tm=m;
        bool f=false;
        for (i=1; i<=n; i++)
        {
            pre[i]=i;
            dp[i]=-1;
            edge[i].clear();
        }
        while (m--)
        {
            scanf("%d%d%d",&a,&b,&c);
            if(f)
                continue;

            int fa=Find(a);
                int fb=Find(b);
                if (fa==fb)
                {
                    f=true;
                    continue;
                }
                else
                {
                    pre[fb]=fa;
                    edge[a].push_back(Edge(b,c));
                    edge[b].push_back(Edge(a,c));
                }


        }

        //while (m--){scanf("%d%d%d",&a,&b,&c);}


        if (f)
        {
            printf("YES\n");
        }
        else
        {
            ans=0;
            for (i=1; i<=n; i++)
            {
                if (dp[i]==-1)
                    dfs(i,-1);
            }
            printf("%d\n",ans);
        }

    }
    return 0;
}