hdu1863 暢通工程(最小生成樹之prim)

Problem Description
省政府“暢通工程”的目標是使全省任何兩個村莊間都可以實現公路交通（但不一定有直接的公路相連，只要能間接通過公路可達即可）。經過調查評估，得到的統計表中列出了有可能建設公路的若干條道路的成本。現請你編寫程序，計算出全省暢通需要的最低成本。

Input
測試輸入包含若干測試用例。每個測試用例的第1行給出評估的道路條數 N、村莊數目M ( < 100 )；隨後的 N
行對應村莊間道路的成本，每行給出一對正整數，分別是兩個村莊的編號，以及此兩村莊間道路的成本（也是正整數）。為簡單起見，村莊從1到M編號。當N為0時，全部輸入結束，相應的結果不要輸出。

Output
對每個測試用例，在1行裡輸出全省暢通需要的最低成本。若統計數據不足以保證暢通，則輸出“?”。

Sample Input
3 3
1 2 1
1 3 2
2 3 4
1 3
2 3 2
0 100

Sample Output
3
?

Source

#include<stdio.h>   
int s[105],node[105],map[105][105],m,sum,INF=10000000;  
void set_first()  
{  
    for(int i=1;i<=m;i++)  
    {  
        s[i]=0; node[i]=INF;  
        for(int j=1;j<=m;j++)  
        map[i][j]=INF;  
    }  
}  
int Prim(int n)  
{  
    int min,t=1;  
    s[n]=1; sum=0;  
    for(int k=2;k<=m;k++)  
    {  
        for(int i=1;i<=m;i++)  
        if(s[i]==0&&node[i]>map[n][i])  
        node[i]=map[n][i];  
  
        min=INF;  
        for(int j=1;j<=m;j++)  
        if(s[j]==0&&min>node[j])  
        {  
            min=node[j]; n=j;  
        }  
        if(s[n]==0)  
        {  
            t++; sum+=min; s[n]=1;  
        }  
    }  
    if(t==m)  
    return 1;  
    return 0;  
}  
int main()  
{  
    int a,b,p,n,t;  
    while(scanf("%d%d",&n,&m)>0&&n)  
    {  
        set_first();  
        for(int i=1;i<=n;i++)  
        {  
            scanf("%d%d%d",&a,&b,&p);  
            if(map[a][b]>p)  
            map[a][b]=map[b][a]=p;  
        }  
  
       t=Prim(1);  
        if(t==1)  
        printf("%d\n",sum);  
        else  
        printf("?\n");  
    }  
}