题目1(prim算法)
收获
1:学到了关于memset(vis,0,sizeof(vis));初始化函数的使用,只有设置成0,才是都初始化为0,其他的如果写10,并不会像我们理想状态都设置为10 的(参考链接),所以这里想要用最大距离作为判断的时候,直接写INT_MAX,会都赋值为-1,这里采用,都设置成0x3f的方法,最终和它define的INF进行比较,若相等则代表并不能生成最小生成树~
2:关于vectora作为函数参数的调用方法自己还是不是很会,只能设置成全局变量,希望在未来的学习中可以知道是为什么?
代码
输入:
6 9
0 1 34
0 2 46
0 5 19
1 4 12
5 4 26
2 5 25
2 3 17
3 5 25
3 4 38
运行截图:
如果对应的没有最小生成树的话,就会输出最大值
#include <iostream>
#include <bits/stdc++.h>
using namespace std;
typedef pair<int,int> pii;
vector<pii>a[100];
#define INF 0x3f3f3f3f
int prim(int n,int dist[],int vis[]){
dist[0]=0;
long long sum=0;
for(int i=0;i<n;i++){
int cur=-1;
for(int j=0;j<n;j++)
if(!vis[j]&&(cur == -1||dist[j]<dist[cur]))
cur=j;
if(dist[cur]>=INF)
return INF;
cout<<dist[cur]<<endl;
sum+=dist[cur];
vis[cur]=1;
for(auto it=a[cur].begin();it!=a[cur].end();it++){
if(!vis[it->first])
{dist[it->first]=min(dist[it->first],it->second);
}
}
}
return sum;
}
int main()
{
int n,m;
cin>>n>>m;
int dist[n];
int vis[n];
memset(vis,0,sizeof(vis));
memset(dist,0x3f,sizeof(dist));
while(m--){
int x,y,w;
cin>>x>>y>>w;
a[x].push_back(make_pair(y,w));
a[y].push_back(make_pair(x,w));
}
cout << prim(n,dist,vis)<< endl;
return 0;
}
题目2(Kruskal算法)
收获
代码
参考链接:https://blog.csdn.net/qq_43619271/article/details/109091314