POJ 2421(prim)

 1 #include <stdio.h>
 2 #include <string.h>
 3 
 4 
 5 #define inf 100009
 6 
 7 bool mark[1001];
 8 int a[1001][1001],dis[1001],ans,n;
 9 
10 int prim()
11 {
12    for(int i=1;i<=n;i++)
13     dis[i]=inf;dis[1]=0;
14     for(int i=1;i<=n;i++){
15         int tep=inf;int k=0;
16         for(int j=1;j<=n;j++){
17             if(mark[j]&&dis[j]<tep)
18             {
19                 tep=dis[j];
20                 k=j;
21             }
22         }
23         if(tep==inf) return 0;
24         ans+=tep;
25         mark[k]=false;
26         for(int j=1;j<=n;j++)
27             if(mark[j]&&dis[j]>a[k][j])
28                 dis[j]=a[k][j];
29        }
30    return 0;
31 }
32 
33 int main()
34 {
35     while(scanf("%d",&n)!=EOF)
36     {
37         int x=0,c,b;
38         memset(mark,true,sizeof(mark));
39         ans=0;
40         for(int i=1;i<=n;i++)
41             for(int j=1;j<=n;j++)
42                 scanf("%d",&a[i][j]);
43         scanf("%d",&x);
44         while(x--)
45         {
46             scanf("%d%d",&b,&c);
47             a[b][c]=0;
48             a[c][b]=0;
49         }
50         prim();
51         printf("%d\n",ans);
52     }
53 }