1. 程式人生 > >P1546 最短網路 Agri-Net

P1546 最短網路 Agri-Net

題目背景

農民約翰被選為他們鎮的鎮長!他其中一個競選承諾就是在鎮上建立起網際網路,並連線到所有的農場。當然,他需要你的幫助。

題目描述

約翰已經給他的農場安排了一條高速的網路線路,他想把這條線路共享給其他農場。為了用最小的消費,他想鋪設最短的光纖去連線所有的農場。

你將得到一份各農場之間連線費用的列表,你必須找出能連線所有農場並所用光纖最短的方案。每兩個農場間的距離不會超過100000

輸入輸出格式

輸入格式:

第一行: 農場的個數,N(3<=N<=100)。

第二行..結尾: 後來的行包含了一個N*N的矩陣,表示每個農場之間的距離。理論上,他們是N行,每行由N個用空格分隔的陣列成,實際上,他們限制在80個字元,因此,某些行會緊接著另一些行。當然,對角線將會是0,因為不會有線路從第i個農場到它本身。

輸出格式:

只有一個輸出,其中包含連線到每個農場的光纖的最小長度。

輸入輸出樣例

輸入樣例#1:

4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0

輸出樣例#1:

28

分析:很明顯是求最小生成樹問題,採用比較簡單的克魯斯卡爾演算法,注意該演算法需要用到並查集來判斷是否形成環路。

一開始陣列開闢空間有點小,導致三個測試點過不去,仔細看了 看發現是陣列定義問題。

AC程式碼:

#include<bits/stdc++.h>
using namespace std;
#define maxn 10005//一半就夠了
typedef struct{
    int x,y,w;
}MAP;
bool cmp(MAP a,MAP b)
{
    return a.w<b.w;
} 
MAP a[maxn];
int f[maxn];
int findx(int x)
{
    int r=x;
    while(f[r]!=r)
    r=f[r];
    return r;
}
int main()
{
int n;
cin>>n;
int k,m=0;
for(int i=1;i<=n;i++)
{
    f[i]=i;
    for(int j=1;j<=n;j++)
    {
        cin>>k;
        if(j>i)//避免重複 
        {
            m++;
            a[m].x=i;
            a[m].y=j;
            a[m].w=k;
        }
    }
}
sort(a+1,a+m+1,cmp);
int ans=0,p=0;
for(int i=1;i<=m;i++)
{
    if(findx(a[i].x)!=findx(a[i].y))
    {
        ans+=a[i].w;
        f[findx(a[i].x)]=a[i].y;
        p++;
        if(p==n-1)
        break;		
    }
}
cout<<ans<<endl;
return 0;	
}