最小生成树 poj 2485

最新推荐文章于 2024-04-04 17:19:45 发布

原创最新推荐文章于 2024-04-04 17:19:45 发布 · 244 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#input #include #im

图算法专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一个使用Kruskal算法解决最小生成树问题的C++实现案例。通过输入顶点数量和边的权重，程序能够找到权重最大的边，该边属于最小生成树。涉及并查集数据结构用于避免形成环路。

#include <cstdio>
#include <iostream>
#include <cmath>
#include <algorithm>
#include <cstdlib>
#include <cctype>
#define M 505
#define IM 505*505/2

using namespace std;

int r[IM];
int w[IM];
int u[IM];
int v[IM];
int p[M];
int cnt;
int n; //ding dian shu

int cmp(const void *a,const void *b)
{
return w[*(int *)a] - w[*(int *)b];
}

void input_deal()
{
int temp;
cin>>n;
cnt = 0;
for(int i = 1;i <= n;i++)
for(int j = 1;j <= n;j++){
cin>>temp;
if(j > i){
w[cnt] = temp;
u[cnt] = i;
v[cnt] = j;
cnt++;
}
}
}

int find(int x)
{
return p[x] == x ? x : p[x] = find(p[x]);
}

int Kruskal()
{
int imax = 0;
for(int i = 0;i <= n;i++) p[i] = i; //初始化并查集
for(int i = 0;i < cnt;i++) r[i] = i; //初始化边序号

qsort(r,cnt,sizeof(r[0]),cmp); //按权值给边排序
for(int i = 0;i < cnt;i++){
int e,x,y;
e = r[i];
x = find(u[e]);
y = find(v[e]);
if(x != y){
if(w[e] > imax) imax = w[e];
p[x] = y;
}
}
return imax;
}

int main()
{
//freopen("in.in","r",stdin);
int cas;
cin>>cas;
while(cas--)
{
input_deal();
int imax = Kruskal();
cout<<imax<<endl;
}
return 0;
}