script merge con 暫時 簡單 content ane str input

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 Hint Huge input, scanf is recommended.

---

思路：最小生成樹模板題。兩種算法都可以，直接上代碼。

坑點：（C++過，G++超時//暫時不知道為啥）

---

克魯斯卡爾算法解：

#include<cstdio>
#include<algorithm>
using namespace std;
int bin[205],n;
int flag[205];
struct node{
    int x,y,val; 
}a[10005];

int cmp(node x,node y){
    if(x.val!=y.val) return
 
 x.val<y.val;
}

int find(int x) {
    if(bin[x] == x)
        return bin[x];
    return bin[x] = find(bin[x]);
}
void merge(int x,int y){
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy)
    bin[fx]=fy;
}
int same(int x,int y){
    if(find(x)==find(y))  return 1;
    return 0;
}

int main(){
    
 
while(scanf("%d",&n)&&n){
        int i;
        int m=n*(n-1)/2;
        for(i=1;i<=n;i++){
            bin[i]=i;
            flag[i]=0;
        }
        for(i=0;i<m;i++)
            scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].val);
        sort(a,a+m,cmp);
         int sum=0,cnt=0;
        for(i=0;i<m;i++){
            if(cnt==n-1) break;        
            if(same(a[i].x,a[i].y))continue;

            sum+=a[i].val;
            merge(a[i].x,a[i].y);
            cnt++;
        }
        printf("%d\n",sum);
    }
    return 0;
}

Prim算法解：

#include<cstdio>
#include<algorithm>
using namespace std;

struct node{
   int x,y,val;
}a[5005];

int cmp(node x,node y){
    if(x.val!=y.val) return x.val<y.val;
}

int main(){
    int i,sum,n,m,num;
    int flag[200];//標記數組表示，目前節點在集合當中。 
    while(scanf("%d",&n),n)
    {
       m=(n*(n-1))/2;
       for(i=0;i<m;i++)
       {
          scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].val);
       }
       sort(a,a+m,cmp);//按照len排序 
       for(i=1;i<=n;i++) flag[i]=0;//初始化標記 
       flag[a[0].x]=1;//標記第一個節點 
       sum=0;
       for(i=0;i<m;i++)
       {
          num=flag[a[i].x]+flag[a[i].y];
           if(num==1)//如果兩個都沒標記肯定不在這個集合裏，所以不操作，如果兩個都標記了說明兩個都在集合裏了，則不用修路了 
           {
             sum+=a[i].val;
             flag[a[i].x]=flag[a[i].y]=1;
             i=0;//可能有漏掉的值小的路需要從頭開始找 （當num==0時就會被漏掉） 
          }
       }
       printf("%d\n",sum);
    }
    return 0;
}

HDU-1233-還是暢通工程（最小生成樹）