最小生成树——城市公交网建设问题

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 using namespace std;
 5 
 6 const int maxn=0x7f;
 7 bool visit[101];
 8 int dis[101];
 9 int map[101][101];
10 int n,m,u,v,h,k;
11 int min1; 
12 
13 void sc(int s)
14 {
15     for(int i=1;i<=n;i++)
16         dis[i]=map[s][i];
17     visit[s]=true;
18     dis[s]=0;
19     for(int i=1;i<=n;i++)
20     {
21         min1=maxn;
22         k=s;
23         for(int j=1;j<=n;j++) 
24         {
25             if(!visit[j]&&dis[j]<min1)
26             {
27                 min1=dis[j];
28                 k=j;
29             }
30         }
31         visit[k]=1;
32         for(int j=1;j<=n;j++)
33         {
34             if(!visit[j]&&map[k][j]<dis[j])
35              dis[j]=map[k][j];
36         }
37     }
38     for(int i=1;i<=n;i++)
39       for(int j=1;j<=n;j++)
40       if(map[i][j]==dis[j])
41          cout<<i<<" "<<j<<endl;
42 }
43 
44 int main()
45 {
46     cin>>n>>m;
47     memset(map,maxn,sizeof(map));
48     for(int i=1;i<=m;i++)
49     {
50         cin>>u>>v>>h;
51         map[u][v]=map[v][u]=h;
52     }
53     for(int i=1;i<=m;i++)
54         dis[i]=maxn;
55     sc(1);
56     return 0;
57 }