#include <bits/stdc++.h>
using namespace std;
//邻接矩阵
int adj_mat[10010][10010];
//简单的来说,dis[i]
//表示编号为i的节点,从已标记的点走i点到花费的最小费用就是dis[i],如果有,那就是被别的节点占用了
//因为最小生成树是在一个连通图的条件下,基于贪心算法,dis加起来就是最小生成树的权值
int dis [10010];
bool s[10010];
int n,m;
int prim(int pos)
{
/*从顶点开始,初始化*/
dis[pos]=0;
int sum=0;
/*一共有m个城市,所以进行m次循环,每次标记一个点,正好m次标记完,最小生成树创建完毕*/
for(int i=1;i<=m;i++){
int min_num=INT_MAX;
int flag;
/*找到已经标记的点到剩余未标记的点的最小权值点,并且记录下标*/
/*dis[j]表示图中所有的点到当前j号节点的最短路径,也就是最小权值*/
/*如果全为false的情况,那么就选中顶点*/
for(int j=1;j<=m;j++){
if(!s[j]&&dis[i]<min_num){
flag=j;
min_num=dis[j];
}
}
/*把这个点变为标记的点*/
s[flag]=true;
/*如果不是一个连通图,就返回*/
if(dis[flag]==0x7f7f7f7f) return 0x7f7f7f7f;
/*更新sum*/
/*dis其实就是每次找最优边的权值*/
sum+=dis[flag];
/*更新dis[i]的值*/
for(int k=1;k<=m;k++){
if(!s[k]){
/*如果点没有被标记
当前点的dis从他当前权值和最优点到当前点的权值选一个较小的*/
dis[k]=min(dis[k],adj_mat[flag][k]);
}
}
}
cout<<"dis 数组中的元素为"<<endl;
for(int i=0;i<=m;i++) {
cout << dis[i] << " ";
}
return sum;
}
int main()
{
memset(adj_mat,0x7f,sizeof(adj_mat));
memset(dis,0x7f,sizeof(dis));
int c;
cin>>c>>n>>m;//n条道路m个城市
//邻接矩阵的初始化
for(int i=1;i<=m;i++){
adj_mat[i][i]=0;
}
//邻接矩阵的输入
for(int i=1;i<=n;i++){
int x,y,z;
cin>>x>>y>>z;
adj_mat[x][y]=min(adj_mat[x][y],z);
adj_mat[y][x]=min(adj_mat[y][x],z);
}
/*prim算法*/
int sum=prim(1);
//cout<<"sum= "<<sum<<endl;
if(sum>c) cout<<"No"<<endl;
else cout<<"Yes"<<endl;
return 0;
}
prim最短生成树——数组模拟
最新推荐文章于 2024-09-06 10:38:06 发布