1. 程式人生 > >POJ 1258 Agri-Net(最小生成樹 Prim 模版題)

POJ 1258 Agri-Net(最小生成樹 Prim 模版題)

大意:新鎮長競選宣言就是將網路帶到每一個農場,給出農場個數,兩兩之間建光纜的耗費,求所有都聯通的最小耗費。

思路:最小生成樹,因為邊比較稠密,用Prim做。

PS;對於比較稠密的圖,用Prim,對於比較稀疏的圖,用 Kruskal。Kruskal是找邊的過程,稀疏的話會比較快。

#include <stdio.h>
#include <string.h>
#define INF 0x3f3f3f3f

int dis[110];
int Map[110][110];
int n;
int Ans;

int min(int a, int b)
{
    return a > b ? b : a;
}

void  Prim()
{
    int Min_ele, Min_node;
    memset(dis, INF, sizeof(dis));
    Ans = 0;
    int r = 1;
    for(int i = 1; i < n; i++)
    {
        dis[r] = -1;
        Min_ele = INF;
        for(int j = 1; j <= n; j++)
        {
            if(dis[j] >= 0)
            {
                dis[j] = min(dis[j], Map[r][j]);
                if(dis[j] < Min_ele)
                {
                    Min_ele = dis[j];
                    Min_node = j;
                }
            }
        }
        r = Min_node;
        Ans += Min_ele;
    }
}


void Solve()
{
    while(~scanf("%d", &n))
    {
        memset(Map, 0, sizeof(Map));
        for(int i = 1; i <= n; i++)
        {
            for(int j = 1; j <= n; j++)
            {
                scanf("%d", &Map[i][j]);
            }
        }
        Prim();
        printf("%d\n", Ans);
    }
}

int main()
{
    Solve();

    return 0;
}