POJ——T2421 Constructing Roads

 1 #include <algorithm>
 2 #include <cstdio>
 3 
 4 using namespace std;
 5 
 6 const int N(1100);
 7 int n,q,u,v,w;
 8 int dis[N][N],ans;
 9 int minn,vis[N],d[N];
10 
11 void Prime()
12 {
13     for(int i=2;i<=n;i++) d[i]=dis[1][i];
14     d[1]=vis[1]=0;
15     for(int i=1;i<n;i++)
16     {
17         minn=0;
18         for(int j=1;j<=n;j++)
19             if(!vis[j]&&(!minn||d[j]<d[minn])) minn=j;
20         vis[minn]=1;
21         for(int j=1;j<=n;j++)
22             if(!vis[j]) d[j]=min(d[j],dis[minn][j]);
23     }
24     for(int i=1;i<=n;i++) ans+=d[i];
25 }
26 
27 int main()
28 {
29     scanf("%d",&n);
30     for(int i=1;i<=n;i++)
31       for(int j=1;j<=n;j++)
32           scanf("%d",&dis[i][j]);
33     scanf("%d",&q);
34     for(;q;q--)
35     {
36         scanf("%d%d",&u,&v);
37         dis[u][v]=dis[v][u]=0;
38     }
39     Prime();
40     printf("%d",ans);
41     return 0;
42 }