pku 1285

1. #include <iostream> 
2. #include <cstring> 
3. const int MAX=101; 
4. const int INF=100000; 
5. using namespace std; 
7. void prim(int matrix[][MAX],int N,int u)  
8. {
9.     int i,j,k;
10.     int min;
11.     int mincost=0; 
12.     int lowcost[MAX]; 
14.     for(i=0;i<N;i++) 
15.         lowcost[i]=matrix[u][i]; 
17.     for(i=1;i<N;i++) 
18.     { 
19.         min=INF;
20.         for(j=0;j<N;j++) 
21.             if(lowcost[j] && lowcost[j]<min)  
22.             {  
23.                 min=lowcost[j];  
24.                 k=j;  
25.             } 
26.         mincost+=min; 
27.         lowcost[k]=0; 
28.         for(j=0;j<N;j++) 
29.             if(lowcost[j] && lowcost[j]>matrix[k][j]) 
30.                 lowcost[j]=matrix[k][j]; 
31.     } 
32.     printf("%d/n",mincost); 
33. } 
35. int main() 
36. {  
37.     int N;  
38.     int i,j; 
39.     int matrix[MAX][MAX]; 
41.     while(cin >> N) 
42.     {
43.         for(i=0;i<N;i++) 
44.             for(j=0;j<N;j++) 
45.                 scanf("%d",&matrix[i][j]); 
46.         prim(matrix,N,0); 
47.     } 
48.     return 0; 
49. }