关闭

NYOJ-38-布线问题(普利姆)

标签: 普利姆
192人阅读 评论(0) 收藏 举报
分类:

布线问题
时间限制:1000 ms | 内存限制:65535 KB
难度:4
描述
南阳理工学院要进行用电线路改造,现在校长要求设计师设计出一种布线方式,该布线方式需要满足以下条件:
1、把所有的楼都供上电。
2、所用电线花费最少
输入
第一行是一个整数n表示有n组测试数据。(n<5)
每组测试数据的第一行是两个整数v,e.
v表示学校里楼的总个数(v<=500)
随后的e行里,每行有三个整数a,b,c表示a与b之间如果建铺设线路花费为c(c<=100)。(哪两栋楼间如果没有指明花费,则表示这两栋楼直接连通需要费用太大或者不可能连通)
随后的1行里,有v个整数,其中第i个数表示从第i号楼接线到外界供电设施所需要的费用。( 0

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<string.h>
#include<math.h>
using namespace std;
const int maxn=505;
const int INF=0x3f3f3f3f;
int map[maxn][maxn];
int dis[maxn];
int vis[maxn];
int N;//N组测试数据
int V;//点的数量
int E;//边的数量
int min_num;//最小外接花费
void init()
{
    scanf("%d%d",&V,&E);
    for(int i=0; i<=V; i++)//初始化图
        for(int j=0; j<=V; j++)
            i==j?map[i][j]=0:map[i][j]=INF;
    while(E--)//建图
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        map[a][b]=c;//无向图
        map[b][a]=c;
    }
    min_num=INF;
    for(int i=1; i<=V; i++)
    {
        int x;
        scanf("%d",&x);
        min_num=min(min_num,x);//生成树连接到外界的最小花费
        dis[i]=map[i][1];//从1开始构造最小生成树
    }
    memset(vis,0,sizeof(vis));
    vis[1]=1;//0代表是外界
}
void Prim()
{
    int min_cost=0;//初始化最小花费
    for(int i=1; i<V; i++)
    {
        int minn=INF;
        int point_minn;
        for(int j=1; j<=V; j++)
            if(vis[j]==0&&minn>dis[j])
            {
                point_minn=j;
                minn=dis[j];
            }
        if(minn==INF)
            break;
        min_cost+=dis[point_minn];
        vis[point_minn]=1;
        for(int j=1; j<=V; j++)
            if(vis[j]==0&&dis[j]>map[point_minn][j])
                dis[j]=map[point_minn][j];
    }
    printf("%d\n",min_cost+min_num);
}
int main()
{
    scanf("%d",&N);
    while(N--)
    {
        init();
        Prim();
    }
    return 0;
}
0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:91056次
    • 积分:4971
    • 等级:
    • 排名:第5647名
    • 原创:414篇
    • 转载:6篇
    • 译文:0篇
    • 评论:42条
    最新评论