prim最短生成树——数组模拟

#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;
}