最优灌溉

问题描述

　　雷雷承包了很多片麦田，为了灌溉这些麦田，雷雷在第一个麦田挖了一口很深的水井，所有的麦田都从这口井来引水灌溉。
　　为了灌溉，雷雷需要建立一些水渠，以连接水井和麦田，雷雷也可以利用部分麦田作为“中转站”，利用水渠连接不同的麦田，这样只要一片麦田能被灌溉，则与其连接的麦田也能被灌溉。
　　现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用（注意不是所有麦田之间都可以建立水渠）。请问灌溉所有麦田最少需要多少费用来修建水渠。

输入格式

　　输入的第一行包含两个正整数n, m，分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。
　　接下来m行，每行包含三个整数a_i, b_i, c_i，表示第a_i片麦田与第b_i片麦田之间可以建立一条水渠，所需要的费用为c_i。

输出格式

　　输出一行，包含一个整数，表示灌溉所有麦田所需要的最小费用。

样例输入

4 4
1 2 1
2 3 4
2 4 2
3 4 3

样例输出

6

样例说明

　　建立以下三条水渠：麦田1与麦田2、麦田2与麦田4、麦田4与麦田3。

评测用例规模与约定

　　前20%的评测用例满足：n≤5。
　　前40%的评测用例满足：n≤20。
　　前60%的评测用例满足：n≤100。
　　所有评测用例都满足：1≤n≤1000，1≤m≤100,000，1≤c_i≤10,000。

这是一道经典的最小生成树的应用问题，在该题的算法实现中，我用到了动态数组，其实也没有必要，不过对算法的实现没有影响，可能会影响到程序的运行效率吧。有心的同志完全可以改写一下。

给源码：

#include <iostream>
using namespace std;
const int MAXCOST = 99999;
bool prime(int **matrix, int n,int& count,int& lowcost)
{
    int *flag = new int[n + 1];
    int index;
    int curlowCost = MAXCOST;
    int *cost = new int[n + 1];
    for (int i = 0; i <= n; i++)
    {
        cost[i] = MAXCOST;
    }

    //初始化
    flag[1] = 1;
    for (int i = 2; i <= n; i++)
    {
        cost[i] = matrix[1][i];
        flag[i] = 0;
    }

    for (int j = 2; j <= n; j++)
    {
        curlowCost = MAXCOST;
        for (int i = 1; i <= n; i++)
        {
            if (cost[i] < curlowCost && flag[i] == 0)
            {
                curlowCost = cost[i];
                index = i;
            }
        }

        flag[index] = 1;
        lowcost += curlowCost;
        for (int k = 1; k <= n; k++)
        {
            if (flag[k] == 0 && matrix[index][k] < cost[k])
                cost[k] = matrix[index][k];
        }
    }

    for (int i = 1; i <= n; i++)
    {
        if (flag[i] == 1)
            count++;
    }
    delete[] flag;

    if (count == n)
        return true;
    else
        return false;

}

void main(void)
{
    int m, n;
    int row, col, val;
    while (cin >> n>>m){
        int count = 0, lowcost = 0;
        int **matrix;
        matrix = new int*[n+1];
        for (int i = 0; i <= n; i++)
        {
            matrix[i] = new int[n+1];
        }

        for (int i = 0; i <= n;i++)
            for (int j = 0; j <= n; j++)
            {
                matrix[i][j] = MAXCOST;
            }

        for (int i = 1; i <= m; i++)
        {
            cin >> row >> col >> val;
            matrix[row][col] = matrix[col][row] = val;
        }

        if (prime(matrix, n, count, lowcost))
            cout << lowcost << endl;
        else
            cout << "error";

        for (int i = 0; i <= n; i++)
            delete[] matrix[i];

    }

}