hdu oj 1102(prim)

分析:裸的最小生成树,没什么好说的.

#include<iostream>
#include<string.h>
#include<math.h>
#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define N 110

using namespace std;

int n,ans;
int Map[N][N],dis[N],vis[N];

void Prim(){
    
    //从1这个节点开始扫,开始初始化 
    for(int i=1;i<=n;i++){
        dis[i]=Map[1][i];
        vis[i]=0;
    }
    dis[1]=0;
    vis[1]=1;
    int k,tmp;
    for(int i=1;i<=n;i++){
        tmp=INF;
        
        //寻找离树最近的点 
        for(int j=1;j<=n;j++)
            if(!vis[j] && tmp>dis[j]){
                k=j;
                tmp=dis[j];
            }
            
        //提前退出,证明剩下的点不连通了    
        if(tmp==INF)
            break;
            
        vis[k]=1;
        ans+=dis[k];
        
        //对各个边进行松弛操作 
        for(int j=1;j<=n;j++)
            if(!vis[j] && dis[j]>Map[k][j])
                dis[j]=Map[k][j];
    }
}

int main()
{
    //while(~scanf("%d",&n)
    while(cin>>n)
	{
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                scanf("%d",&Map[i][j]);
        int q,a,b;
        cin>>q;
        while(q--){
            cin>>a>>b;
            Map[a][b]=Map[b][a]=0;
        }
        ans=0;
        Prim();
        cout<<ans<<endl;
    }
    return 0;
}
发布了183 篇原创文章 · 获赞 31 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/mlm5678/article/details/104666162