【图-最小生成树】NOIP2010 模拟试题 新的开始

【题目描述】
发展采矿业当然首先得有矿井， 小 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） 。
【输出格式】
仅一个整数， 表示让所有矿井获得充足电能的最小花费。

【题目分析】
乍一看似乎毫无头绪，但是根据其最小生成树的本质，可以将发电站转换为边的存在，即添加一个0号节点，连接所有节点，然后跑最小生成树即可。

#include <iostream>
#include <cstdio>
#include <climits>
using namespace std;
const int N=301;
const int INF=INT_MAX;
int n;   long long ans;
int map[N][N],close[N],least[N];  bool used[N];
void read()
{
    int i,j;
    scanf("%d",&n);
    for (i=1;i<=n;i++)
    {
        scanf("%d",&map[0][i]);
        map[i][0]=map[0][i];
    }
    for (i=1;i<=n;i++)
        for (j=1;j<=n;j++)
        scanf("%d",&map[i][j]);
    for (i=0;i<=n;i++)
    {
        least[i]=map[0][i];
        close[i]=0;
    }
    used[0]=1;
    return;
}
void get_tree()
{
    int i,mymin,j,now;
    for (i=1;i<=n;i++)
    {
        mymin=INF;
        for (j=0;j<=n;j++)
            if (least[j]<mymin)
            if (!used[j])
            {
                mymin=least[j];
                now=j;
            }
        used[now]=1;
        for (j=0;j<=n;j++)
        if (!used[j])
            if (map[j][now]<least[j])
            {
                least[j]=map[j][now];
                close[j]=now;
            }
    }
    return;
}
void work()
{
    int i;
    for (i=1;i<=n;i++)
        ans+=map[i][close[i]];
    printf("%lld",ans);
    return;
}
int main()
{
    freopen("Water.in","r",stdin);
    freopen("Water.out","w",stdout);
    read();
    get_tree();
    work();
    return 0;
}