学了最小生成树prim算法:
从某一个顶点开始构建生成树;
每次将代价最小的新顶点纳入生成树,直到所有顶点都纳入为止。
实现的算法思想:从Vo开始,总共需要 n-1 轮处理每一轮处理:循环遍历所有个结点,找到lowCast最低的,且还没加入树的顶点。再次循环遍历,更新还没加入的各个顶点的lowCast值
kruskal算法:
每次选择一条权值最小的边,条边的两头连通(原本已经连通的就不选)
直到所有结点都连通
看一道题目:P3366 【模板】最小生成树 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
思路:
0.任取一点
1.找不在最小生成树的高最小生成树最近的点k
2.将k加入最小生成树中
3.更新其他店到最小生成树的距离,重复直到所有点进入最小生成树
代码:
#include <iostream>
#include <bits/stdc++.h>
#include <cstdio>
#include <vector>
using namespace std;
#include <cstring>
struct node
{
int v,w;//v为到达的点,w为边全值
};
vector<node>g[5005];
int dis[5005];//dis[i]:从结点i出发到最小生成树的最近距离
bool vis[5005];//vis[i] true/false i在/不在最小生成树中;
int n,m;
int prim(int rt)
{
//tr为树根
//0. 初始化
memset(dis,0x3f,sizeof(dis));
dis[rt]=0;
int sum=0;//权值和
int cnt=0;//最小生成树结点个数
for(int i=1; i<=n; i++)
{
//1. 寻找不在最小生成树中离最小生成树最近的点
int k=0;
for(int j=1; j<=n; j++)
{
if(!vis[j]&&dis[j]<dis[k])
{
k=j;
}
}
if(k==0)
{
break;
}
//找到k,将k加入最小生成树中
vis[k]=1;
sum+=dis[k];
cnt++;
//3.更新其他点离最小生成树
//遍历k的邻接点
for(int j=0; j<g[k].size(); j++)
{
int v=g[k][j].v,w=g[k][j].w;
if(!vis[v]&&w<dis[v])
{
dis[v]=w;
}
}
}
if(cnt==n)
{
return sum;
}
else
{
return -1;
}
}
int main()
{
int x,y,z;
scanf("%d%d",&n,&m);
for(int i=1; i<=m; i++)
{
scanf("%d%d%d",&x,&y,&z);
g[x].push_back(node{y,z});
g[y].push_back(node{x,z});
}
int ans=prim(1);
if(ans==-1)
{
printf("orz\n");
}
else
{
printf("%d\n",ans);
}
return 0;
}