ZOJ 1586 prim算法求最小生成树

题意：建立QS网络的最小费用。

分析：运用prim算法求解最小生成树

关键：本题关键是建图。在构造有向网时，每条边的权值为两个ＱＳ的适配器的加个加上两个QS之间网线的加个。这样就可以解决问题了。

简化：本题只需最小生成树的权值，不需要记录构造最小生成树时选择的边，因此可以将lowcost数组和nearvex数组合二为一。

自己做法还是一起解决吧，省得记那么多。呵呵

// I'm the Topcoder
//C
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <math.h>
#include <time.h>
//C++
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <cctype>
#include <stack>
#include <string>
#include <list>
#include <queue>
#include <map>
#include <vector>
#include <deque>
#include <set>
using namespace std;

//*************************OUTPUT*************************
#ifdef WIN32
#define INT64 "%I64d"
#define UINT64 "%I64u"
#else
#define INT64 "%lld"
#define UINT64 "%llu"
#endif

//**************************CONSTANT***********************
#define INF 0x3f3f3f3f
#define eps 1e-8
#define PI acos(-1.)
#define PI2 asin (1.);
typedef long long LL;
//typedef __int64 LL;   //codeforces
typedef unsigned int ui;
typedef unsigned long long ui64;
#define MP make_pair
typedef vector<int> VI;
typedef pair<int, int> PII;
#define pb push_back
#define mp make_pair

//***************************SENTENCE************************
#define CL(a,b) memset (a, b, sizeof (a))
#define sqr(a,b) sqrt ((double)(a)*(a) + (double)(b)*(b))
#define sqr3(a,b,c) sqrt((double)(a)*(a) + (double)(b)*(b) + (double)(c)*(c))

//****************************FUNCTION************************
template <typename T> double DIS(T va, T vb) { return sqr(va.x - vb.x, va.y - vb.y); }
template <class T> inline T INTEGER_LEN(T v) { int len = 1; while (v /= 10) ++len; return len; }
template <typename T> inline T square(T va, T vb) { return va * va + vb * vb; }

// aply for the memory of the stack
//#pragma comment (linker, "/STACK:1024000000,1024000000")
//end

#define maxn 2000+10
int n,m;
int edge[maxn][maxn];//邻接矩阵
int lowcost[maxn];
int nearvex[maxn];
int sumweight=0;
int value[maxn];
void prim(int u0){
    //从顶点u0出发执行普里姆算法
    sumweight=0;//生成树的权值
    for(int i=1;i<=n;i++){
        //初始化lowcost[]数组和neartxt数组
        lowcost[i]=edge[u0][i];
        nearvex[i]=u0;
    }
    nearvex[u0]=-1;
    for(int i=1;i<n;i++){
        int min=INF;
        int v=-1;
        //在lowcoat数组的nearvex[]值为-1的元素中找最小值
        for(int j=1;j<=n;j++){
            if(nearvex[j]!=-1&&lowcost[j]<min){
                v=j;
                min=lowcost[j];
            }
        }
        if(v!=-1){
            //v==-1表示没找到权值最小的边
           // printf("%d %d %d\n",nearvex[v],v,lowcost[v]);
            nearvex[v]=-1;
            sumweight+=lowcost[v];
            for(int j=1;j<=n;j++){
                if(nearvex[j]!=-1&&edge[v][j]<lowcost[j]){
                    lowcost[j]=edge[v][j];
                    nearvex[j]=v;
                }
            }
        }
    }
    printf("%d\n",sumweight);
}

int main(){
    int w;
    int t;
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        sumweight=0;
        //初始化
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                edge[i][j]=INF;
            }
            edge[i][i]=0;
        }

        for(int i=1;i<=n;i++){
            value[i]=0;
        }

        for(int i=1;i<=n;i++){
            scanf("%d",&value[i]);
        }
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
            scanf("%d",&w);
            edge[i][j]=w;
            edge[i][j]+=value[i]+value[j];
            }
        }
        prim(1);
    }
    return 0;
}