度度熊的王国战略 并查集 百度之星

题目链接:题目传送门
思路:判断是否连通,连通则寻找最小的值,不连通则为零
代码:

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
typedef long long int ll;
const ll N=3100;
int n,m,f[N],sum[N];
void init()
{
 for(int i=0;i<=N;i++)
 {
  f[i]=i;
 }
}
int find(int x)
{
 while(x!=f[x])
 {
  x=f[x];
 }
 return x;
}//路径压缩 
int main()
{
 while(cin>>n>>m)
 {
  for(int i=1;i<=n;i++)
  f[i]=i;
  memset(sum,0,sizeof(sum));//初始化 
  int j=1,x,y;
  for(int i=1;i<=m;i++)
  {
   int a,b,c;
   cin>>a>>b>>c;
   if(a==b)continue;
   x=find(a),y=find(b);
   sum[a]+=c,sum[b]+=c;//叠加 
   if(x!=y)
   {
    j++;//判断是不是连通的元素 
    f[x]=y;
   }
  }
  if(j==n)
  {
   sort(sum+1,sum+1+n);
   printf("%d\n",sum[1]);
  }
  else printf("0\n");
 }
 return 0; 
}

你可能感兴趣的:(数据结构)