腾讯第二届马拉松22号第五题

最新推荐文章于 2025-07-06 17:04:20 发布

kay_zhyu

最新推荐文章于 2025-07-06 17:04:20 发布

阅读量680

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/kay_zhyu/article/details/8708826

本文介绍了一种使用并查集解决图中是否存在环的问题，并给出了完整的代码实现。当图中存在环时输出yes，否则输出最长路径。

题目改天再上吧，最近太忙了，先说说思路，其实这道题并不难，用并查集做，实际上题目抽象了之后就是一张图，如果图中有环，则输出yes，没环就输入最长的路径。

其实跟我另外一篇博客畅通工程的大体思路是一样的，链接：http://blog.youkuaiyun.com/kay_zhyu/article/details/8679268

废话不多说，直接附上参考代码。

#include<stdio.h>

int set[100000+10];
int use[100000+10];

int find(int x)
{
 int r = set[x];
 while(r != set[r])
 {
  r = set[r];
 }
 return r;
}

int merge(int r1, int r2, int a, int b, int w)
{
 int temp;
 temp = use[r2] + use[r1] + w;

 if(r1 < r2)
 {
  set[r2] = r1;
  use[r2] = temp;
  use[r1] = temp;
 }
 else
 {
  set[r1] = r2;
  use[r1] = temp;
  use[r2] = temp;
 }
 return temp;
}

void main()
{
 int n,m;
 int u,v,w;
 int i,temp;
 int nMax,flag;
 int r1,r2;

 while(scanf("%d %d",&n,&m) != EOF)
 {
  for(i = 1; i <= 100001; ++i)
  {
   set[i] = i;
   use[i] = -1;//赋值成-1，一方面作为标记，另一方面，在访问过后，把标记改成0，这样不会影响最长路径的求解。
  }

  nMax = -1;
  flag = 0;
  for(i = 0; i < m; ++i)
  {
   scanf("%d %d %d",&u,&v,&w);
   if(flag == 1)
   {
    continue;
   }
   if(use[u] == -1)
   {
    use[u] = 0;
   }
   if(use[v] == -1)
   {
    use[v] = 0;
   }

   r1 = find(u);
   r2 = find(v);

   if(r1 == r2)
   {   
    flag = 1;
   }
   else
   {
    temp = merge(r1,r2,u,v,w);
    nMax = nMax > temp ? nMax : temp;
   }
  }
  if(flag == 1)
  {
   printf("YES\n");
  }
  else
  {
   printf("%d\n", nMax);
  }
 }
}