閱讀991 返回首頁    go 阿裏雲 go 技術社區[雲棲]


HDU1233還是暢通工程

還是暢通工程

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 21896    Accepted Submission(s): 9768

Problem Description
某省調查鄉村交通狀況,得到的統計表中列出了任意兩村莊間的距離。省政府“暢通工程”的目標是使全省任何兩個村莊間都可以實現公路交通(但不一定有直接的公路相連,隻要能間接通過公路可達即可),並要求鋪設的公路總長度為最小。請計算最小的公路總長度。
 
Input
測試輸入包含若幹測試用例。每個測試用例的第1行給出村莊數目N ( < 100 );隨後的N(N-1)/2行對應村莊間的距離,每行給出一對正整數,分別是兩個村莊的編號,以及此兩村莊間的距離。為簡單起見,村莊從1到N編號。
當N為0時,輸入結束,該用例不被處理。
 
Output
對每個測試用例,在1行裏輸出最小的公路總長度。
 
Sample Input
3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0
 
Sample Output
3

5



方法:(prim算法)

先用快排將所有路徑按照從小到大的順序排列,然後開一個節點標記數組,初始值定為1,將排在第一個的路徑的前一個節點的標記設為0。之後進行計算,凡是前後路徑標記之和為1的,說明之前兩條路是不連通的,將長度加上,如果兩條路之和加上為2,說明這兩個節點之前已經可以走通了(兩個節點中間一定有一個過渡節點,如1,2,和1,3,那麼2,3一定聯通),此時不加長度。每次搜索出來一個節點長度,就要返回從第一個重新開始搜索,以免少計算路徑。直到所有路徑全部搜索完畢,結束。求出的路徑之和即為所求的最短路徑(最小生成樹)

#include<iostream>
#include<algorithm>
using namespace std;
struct node
{
   int a,b;
   int len;
}per[5000];
int cmp(node x,node y)
{
    if(x.len!=y.len) return x.len<y.len;
}
int main()
{
    int i,sum,n,m,num;
    int flag[200];
    while(scanf("%d",&n),n!=0)
    {
       m=(n*(n-1))/2;
       for(i=0;i<m;i++)
       {
          scanf("%d%d%d",&per[i].a,&per[i].b,&per[i].len);
       }
       sort(per,per+m,cmp);
       for(i=1;i<=n;i++) flag[i]=1;
       flag[per[0].a]=0;
       sum=0;
       for(i=0;i<m;i++)
       {
          num=flag[per[i].a]+flag[per[i].b];
          if(num==1)
          {
             sum+=per[i].len;
             flag[per[i].a]=flag[per[i].b]=0;
             i=0;
          }
       }
       printf("%d\n",sum);
    }
    return 0;
}

最後更新:2017-04-03 12:54:57

  上一篇:go Linux下安裝vnstat流量統計
  下一篇:go [LeetCode]28.Implement strStr()