hdu 1102

http://acm.hdu.edu.cn/showproblem.php?pid=1102

把已经修建的城市之间值赋0，然后prim。。。

#include<iostream>#include<string.h>using namespace std;const int MAX=101;const int MAXINT=123456789;int map[MAX][MAX];int visit[MAX];int dist[MAX]; int n,q,num,total;void prim(){ visit[1]=1; for(int i=2;i<=n;i++) dist[i]=map[1][i]; int k,min; for(int i=1;i<n;i++) { min=MAXINT; for(int j=1;j<=n;j++) if(!visit[j]&&dist[j]<min) { min=dist[j]; k=j; } visit[k]=1; total+=min; for(int j=1;j<=n;j++) if(!visit[j]&&map[k][j]<dist[j]) dist[j]=map[k][j]; } }int main(){ while(cin>>n) { total=num=0; memset(visit,0,sizeof(visit)); for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) cin>>map[i][j]; cin>>q; int x1,x2; for(int i=1;i<=q;i++) { cin>>x1>>x2; map[x1][x2]=map[x2][x1]=0; } prim(); cout<<total<<endl; } return 0; }#include<iostream>#include<string.h>using namespace std;const int MAX=101;const int MAXINT=123456789;int map[MAX][MAX];int visit[MAX];int dist[MAX]; int n,q,num,total;void prim(){ visit[1]=1; for(int i=2;i<=n;i++) dist[i]=map[1][i]; int k,min; for(int i=1;i<n;i++) { min=MAXINT; for(int j=1;j<=n;j++) if(!visit[j]&&dist[j]<min) { min=dist[j]; k=j; } visit[k]=1; total+=min; for(int j=1;j<=n;j++) if(!visit[j]&&map[k][j]<dist[j]) dist[j]=map[k][j]; } }int main(){ while(cin>>n) { total=num=0; memset(visit,0,sizeof(visit)); for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) cin>>map[i][j]; cin>>q; int x1,x2; for(int i=1;i<=q;i++) { cin>>x1>>x2; map[x1][x2]=map[x2][x1]=0; } prim(); cout<<total<<endl; } return 0; }