acm hdu 1102 Constructing Roads(kruskal)

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

 

这题折磨了我好久，从上午一直搞到晚上，真是惭愧啊！我也不知道该说些什么了……

我知道这题用prim算法是可以的，写起来也简单，只是我自己一直在用kruskal算法

，不过对kruskal算法也是有了进一步的理解……应该还是值得的~

 

#include<iostream> #include<stdio.h> using namespace std; const int MAX=105; int Dis[MAX][MAX]; int pre[MAX]; int count; struct Value { int val; int vexi; int vexj; }edge[10000];//边的个数会远大于结点个数 //因为一个结点到其余每个结点都有一条边 int Kruskal(int n) { int i,j,temp,flag=0,sum=0; for(i=0;i<count;i++)//这边的i的最大值要是边的个数而不是 {//权值最小的前n-1个，因为如果不是就可能会疏漏一些点。 temp=pre[edge[i].vexi]; if(pre[edge[i].vexi]!=pre[edge[i].vexj]) { sum+=edge[i].val; ++flag; for(j=1;j<=n;j++) {//归并到一个根结点 if(pre[j]==temp) pre[j]=pre[edge[i].vexj]; } } } return sum; } int main() { // freopen("e://test.txt","r",stdin); int N,i,j,vi,vj,k,Q; while(scanf("%d",&N)!=EOF) { for(i=1;i<=N;i++) { for(j=1;j<=N;j++) scanf("%d",&Dis[i][j]); }//读入矩阵 for(i=1;i<=N;i++) pre[i]=i; // for(i=1;i<=N;i++) // cout<<pre[i]<<" "; // cout<<endl; scanf("%d",&Q); for(i=1;i<=Q;i++) {//把已经修好的路记为零，好像以前也有过同样的技巧处理 // 因此如果一旦这样处理那用prim算法也是可以做的，有时间做做 scanf("%d%d",&vi,&vj); Dis[vi][vj]=0; Dis[vj][vi]=0; } k=0; for(i=1;i<=N;i++) { for(j=i+1;j<=N;j++,k++) { edge[k].val=Dis[i][j]; edge[k].vexi=i; edge[k].vexj=j; } } count=k; for(i=1;i<k;i++)//插入排序,或用qsort函数，有时间再试试 { for(j=i;j>0&&edge[j].val<edge[j-1].val;j--) swap(edge[j],edge[j-1]); } cout<<Kruskal(N)<<endl; // for(i=1;i<=N;i++) // cout<<pre[i]<<" "; // cout<<endl; } return 0; }

 

下面是用prim算法实现的，写起来比较简单……

 

#include<iostream> #include<stdio.h> #include<cmath> using namespace std; #define MAXSIZE 100 #define MAX 10000 int Dis[MAXSIZE][MAXSIZE]; int lowcost[MAXSIZE]; void prim(int n) { int i,j,k,min,set[MAXSIZE]; memset(set,0,sizeof(set)); set[0]=1; for(i=0;i<n;i++) lowcost[i]=Dis[0][i]; for(i=1;i<n;i++) { min=MAX; for(j=1;j<n;j++) { if(lowcost[j]<min&&set[j]!=1) { min=lowcost[j]; k=j; } } set[k]=1; for(j=1;j<n;j++) { if(set[j]!=1&&lowcost[j]>Dis[k][j]) lowcost[j]=Dis[k][j]; } } } int main() { freopen("test.txt","r",stdin); int i,j,k,N,Q,sum; while(scanf("%d",&N)!=EOF) { for(i=0;i<N;i++) { for(j=0;j<N;j++) scanf("%d",&Dis[i][j]); } scanf("%d",&Q); for(i=0;i<Q;i++) {//下标从零开始 scanf("%d%d",&j,&k); Dis[j-1][k-1]=0; Dis[k-1][j-1]=0; } prim(N); sum=0; for(i=0;i<N;i++) sum+=lowcost[i]; cout<<sum<<endl; } return 0; }