ZOJ

ZOJ - 1586 QS Network （Prim）

#include<iostream>
#include<cstring>
using namespace std;
const int maxn = 1000+10;
const int MAX = 100000;//无穷远 
int n;
int cost[maxn];
int Edge[maxn][maxn];
int lowcost[maxn];
void Init()
{
    cin>>n;
    for(int i=0;i<n;i++)
    {//读入每个结点的适配器价值 
        cin>>cost[i];
    }
    for(int i=0;i<n;i++)
    {//建立邻接矩阵 
        for(int j=0;j<n;j++)
        {
            cin>> Edge[i][j];
            if(i == j) Edge[i][j] = MAX; 
            else Edge[i][j] += cost[i]+cost[j]; 
        }
    }
    memset(lowcost,0,sizeof(lowcost));
}

void prim()
{
    int sum = 0;
    //从源点0开始
    lowcost[0] = -1;//-1表示当前结点已经加入
    for(int i=1;i<n;i++)
        lowcost[i] = Edge[0][i];
    for(int i=1;i<n;i++)
    {//把其他 n-1 个结点 扩展到生成树中 
        int min = MAX,pos;
        for(int j=0;j<n;j++)//找到权值最小的边 
        {
            if(lowcost[j] != -1 && lowcost[j]<min)
            {
                min = lowcost[j];pos = j;
            }
        }
        //将新顶点加入
        lowcost[pos] = -1;
        sum += min;
        //更新lowcost数组
         for(int k=0;k<n;k++)
         {
             if(lowcost[k] > Edge[pos][k]) 
             lowcost[k] = Edge[pos][k];
         }
     } 
     cout<<sum<<endl; 
}

int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        Init();
        prim();
    }
    return 0;
}