HDU：4514 湫湫系列故事——设计风景线（成环判断+树的最大直径）

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

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

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

题目大意：给你一些节点连接信息，问是否成环，成环输出yes，不成环输出最大链长（注意这里可能多棵树）。

解题思路：最大树直径模板+成环判定。

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
int n,m;
struct node
{
	int from,to,val,last;
}bian[2000010];
int pre[100010];
int dis[100010];
int vis[100010];
int head[100010];
int root[100010];
int rootnum;
int num;
int start;
void add(int s,int e,int power)
{
	bian[num].from=s;
	bian[num].to=e;
	bian[num].val=power;
	bian[num].last=head[s];
	head[s]=num;
	num++;
}
void bfs(int x)
{
	int r=x;
	queue<int>q;
	q.push(r);
	vis[r]=1;
	dis[r]=0;
	while(!q.empty())
	{
		int tmp1=q.front();
		q.pop();
		for(int i=head[tmp1];i!=-1;i=bian[i].last)
		{
			int tmp2=bian[i].to;
			if(vis[tmp2]==0&&dis[tmp2]<dis[tmp1]+bian[i].val)
			{
				dis[tmp2]=dis[tmp1]+bian[i].val;
				vis[tmp2]=1;
				q.push(tmp2);
			}
		}
	}
	int max=0;
	for(int i=1;i<=n;i++)
	{
		if(dis[i]>max)
		{
			max=dis[i];
			start=i;
		}
	}
}
int find(int x)
{
	int r=x;
	while(r!=pre[r])
	{
		r=pre[r];
	}
	int i=x;
	int j;
	while(i!=r)
	{
		j=pre[i];
		pre[i]=r;
		i=j;
	}
	return r;
}
int main()
{
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		for(int i=1;i<=n;i++)
		{
			pre[i]=i;
		}
		int flaghuan=0;
		memset(head,-1,sizeof(head));
		for(int i=0;i<m;i++)
		{
			int x,y,z;
			scanf("%d%d%d",&x,&y,&z);
			if(flaghuan==0)//成环了就不需要链表浪费时间了 
			{
				add(x,y,z);
				add(y,x,z);
			}
			int fx=find(x);
			int fy=find(y);
			if(fx==fy)
			{
				flaghuan=1;
			}
			pre[fx]=fy;
		}
		if(flaghuan==1)
		{
			printf("YES\n");
			continue;
		}
		rootnum=0;
		for(int i=1;i<=n;i++)//统计root个数，可能有多棵树 
		{
			if(pre[i]==i)
			{
				root[rootnum++]=i;//把根节点们打表 
			}
		}
		int ans=0;
		for(int i=0;i<rootnum;i++)//比较哪棵树的单链最长 
		{
			num=0;
			memset(vis,0,sizeof(vis));
			memset(dis,0,sizeof(dis));
			bfs(root[i]);
			memset(vis,0,sizeof(vis));
			memset(dis,0,sizeof(dis));
			bfs(start);
			ans=max(ans,dis[start]);
		}
		printf("%d\n",ans);
	}
	return 0;
}