HDU 4514 湫湫系列故事——设计风景线 （树的直径+并查集）

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

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

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
  
  

 

Source

2013腾讯编程马拉松初赛第二场（3月22日）

 

Recommend

liuyiding   |   We have carefully selected several similar problems for you:   5792  5791  5790  5789  5788 

 

并查集判断环，树的直径找最长距离，（注意找最长的距离时，可能不止一个树。。。这个坑了我好长时间）

#include<stdio.h>
#include<string.h>
#include<queue>
#include<algorithm>
#define MAX 100010
using namespace std;
int pre[MAX],ran[MAX],flag;
struct node  
{  
    int to,val,next;  
}edge[MAX*10];
int head[MAX],dis[MAX];  
bool vis[MAX];  
int top,T,ans;
void init(int n)
{
	ans=-1;
	top=0;
	int i;
	for(i=0;i<=n;i++)
	pre[i]=i;
	memset(ran,0,sizeof(ran));
	memset(head,-1,sizeof(head));
}
int find(int x)
{
	if(x==pre[x])
	return x;
	return pre[x]=find(pre[x]);
}
void join(int x,int y)//加权规则，判断环 
{
	if(x==y) 
	return ;
	if(ran[x]>=ran[y])
	pre[y]=x;
	else
	{
	   if(ran[x]==ran[y])
	   ran[y]++;
	   pre[x]=y;
	}
} 
void add(int u,int v,int w)  
{  
    //edge[top].from=u;  
    edge[top].to=v;  
    edge[top].val=w;  
    edge[top].next=head[u];  
    head[u]=top++;  
}  
void BFS(int s)  
{  
    memset(vis,false,sizeof(vis));  
    memset(dis,0,sizeof(dis));  
    queue<int>q;  
    dis[s]=0;  
    vis[s]=true;  
    q.push(s);  
    T=s;  
    ans=0;  
    while(!q.empty())  
    {  
        int u=q.front();  
        q.pop();  
        for(int i=head[u];i!=-1;i=edge[i].next)  
        {  
            int v=edge[i].to;  
            if(!vis[v])  
            {  
                if(dis[v]<dis[u]+edge[i].val)  
                {  
                    dis[v]=dis[u]+edge[i].val;  
                    if(ans<dis[v])  
                    {  
                        ans=dis[v];  
                        T=v;  
                    }  
                }  
                vis[v]=true;  
                q.push(v);  
            }  
        }  
    }  
}  
int main()
{
	int n,m,a,b,c,i,j;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		bool flag=false;
		init(n);
		for(i=0;i<m;i++)
		{
			scanf("%d%d%d",&a,&b,&c);
			if(!flag)//还不存在环，仍需要判断,否则输入完了不用管
	       { 
				int fa=find(a);
				int fb=find(b);
				if(fa==fb)//存在环了
				{
				    flag=true;
				}
				else
				{
				  join(fa,fb);
				  add(a,b,c);
				  add(b,a,c);
				}
	        }
		}
		if(flag)
		{
			printf("YES\n");
		}
		else
		{
			int num=0;
			for(i=1;i<=n;i++)
			{
				if(pre[i]==i&&head[i]!=-1)//一个树，并且有子树，不是一个点 
				{
					BFS(i);
		            BFS(T);
		            num=max(num,ans);
				}
			}
		    printf("%d\n",num);
		}
	}
	return 0;
}