最小生成树——新的开始

问题 D: 新的开始

时间限制: 1 Sec  内存限制: 128 MB
提交: 5  解决: 2
[提交][状态][讨论版][命题人:add_xiezhenghao]

题目描述

发展采矿业当然首先得有矿井， 小FF花了上次探险获得的千分之一的财富请人在岛上挖了n口矿井， 但他似乎忘记考虑的矿井供电问题…… 为了保证电力的供应， 小FF想到了两种办法：

1. 在这一口矿井上建立一个发电站，费用为v（发电站的输出功率可以供给任意多个矿井）。
2. 将这口矿井与另外的已经有电力供应的矿井之间建立电网， 费用为p。

小FF希望身为”NewBe_One" 计划首席工程师的你帮他想出一个保证所有矿井电力供应的最小花费。

输入

第一行一个整数n， 表示矿井总数。 第2～n+1行，每行一个整数， 第i个数v[i]表示在第i口矿井上建立发电站的费用。 接下来为一个n*n的矩阵P， 其中p[ i , j ]表示在第i口矿井和第j口矿井之间建立电网的费用（数据保证有p[ i, j ] = p[ j, i ], 且 p[ i, i ]=0）。

输出

仅一个整数， 表示让所有矿井获得充足电能的最小花费。

样例输入

4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0

样例输出

9

提示

对于30%的数据，1<=n<=50.

对于100%的数据：1<=n<=300,0<=v[i],p[i,j]<=100000.

将发电站看成是0点到每个矿井的距离，剩下的就是求最小生成树用prim就可以

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#define inf 0x3f3f3f3f
using namespace std;
int n,p[305][305],low[305],visit[305],ans=0;
void prim()
{
    int temp,k;
    memset(visit,0,sizeof(visit));
    memset(low,inf,sizeof(low));
    for(int i=0;i<=n;i++)
        low[i]=p[i][0];
    visit[0]=1;
    for(int i=0;i<=n;i++)
    {
        temp=inf;
        for(int j=0;j<=n;j++)
            if(!visit[j]&&temp>low[j])
            {
                temp=low[j];
                k=j;
            }
        if(temp==inf)break;
        visit[k]=1;
        ans+=low[k];
        for(int j=0;j<=n;j++)
            if(!visit[j]&&p[j][k]<low[j])
            low[j]=p[j][k];
    }
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
        {
            cin>>p[0][i];
            p[i][0]=p[0][i];
        }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            cin>>p[i][j];
    prim();
    cout<<ans<<endl;
    return 0;
}