prim算法

题目信息
运行结果
本题排行
讨论区

布线问题
时间限制：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<stdio.h>#include<vector>#include<cstdlib>#include<cstring>using namespace std;int map[505][505]={0};int prim(int v){    int llong[505]={0};    int dic[505]={0};    int sum=0,pos;    pos=1;    dic[pos]=1;    for(int i=1;i<=v;i++){        llong[i]=map[pos][i];    }    for(int l=1;l<v;l++){        int minn=10000;        for(int i=1;i<=v;i++){            if(llong[i]<minn&&llong[i]&&!dic[i]){                minn=llong[i];                pos=i;            }        }        sum+=minn;        dic[pos]=1;        for(int i=1;i<=v;i++){            if((llong[i]==0||(llong[i]>map[pos][i]&&map[pos][i]))&&!dic[i]){                llong[i]=map[pos][i];            }        }    }    return sum;}int main(){    int k;    scanf("%d",&k);    while(k--){        int v,e;        memset(map,0,sizeof(map));        scanf("%d%d",&v,&e);        int a,b,c,hh;        for(int i=0;i<e;i++){            scanf("%d%d%d",&a,&b,&c);            map[a][b]=c;            map[b][a]=c;        }        int minn=-1;        for(int i=0;i<v;i++){            scanf("%d",&hh);            if(minn==-1)                minn=hh;            else                if(minn>hh)                minn=hh;        }        printf("%d\n",prim(v)+minn);    }    return 0;}