3 3 1 2 1 1 3 2 2 3 4 1 3 2 3 2 0 100
3 ?
本题我犯了一个低级错误,把sort(eg,eg+n,cmp);错写成sort(eg,eg+m,cmp);
#include<stdio.h> #include<string.h> #include<algorithm> using namespace std; int n,m; struct stu{ int u,v,l; }; stu eg[10000]; int h[10000]; int cmp(stu a,stu b) { return a.l<b.l; } int find(int x) { //int r=x; if(x==h[x]) return x; return h[x]=find(h[x]); } int join(int x,int y) { int fx=find(x); int fy=find(y); if(fy!=fx) { h[fx]=fy; return 1; } return 0; } int main() { int i,j,k; while(scanf("%d",&n)&&n) { scanf("%d",&m); for(i=0;i<n;i++) scanf("%d%d%d",&eg[i].u,&eg[i].v,&eg[i].l); sort(eg,eg+n,cmp);//出现错误的地方; if(n<m-1)//这个可以省略但是后面的不能 {// printf("?\n");// continue;// }// int w=0; for(i=1;i<=m;i++) h[i]=i; int count=0; for(i=0;i<n;i++) { if(join(eg[i].u,eg[i].v)) { w+=eg[i].l; count++; } } if(count==m-1)//不可以省略,因为如果省略n大于m时也不一定能达到目的。 printf("%d\n",w); else// printf("?\n");// } return 0; }
再贴一个用prim()算法写出来的代码:
#include<stdio.h> #include<string.h> #include<math.h> #define M 1000000 int eg[400][400],vid[10000]; int a[1000],b[1000],p[1000]; int n,m; void prim() { int k,r; int min; memset(p,0,sizeof(p)); for(int i=1;i<=m;i++) vid[i]=eg[1][i]; p[1]=1;vid[1]=0; for(int i=2;i<=m;i++) { min=M; k=1; for(int j=1;j<=m;j++) if(!p[j]&&vid[j]<min) { min=vid[j]; k=j; } if(min==M){ printf("?\n"); return ; } p[k]=1; for(int t=1;t<=m;t++) if(!p[t]&&vid[t]>eg[k][t]) vid[t]=eg[k][t]; } int sum=0; for(int i=2;i<=m;i++) sum+=vid[i]; printf("%d\n",sum); } int main() { while(scanf("%d",&n)&&n) { int i,j,k; scanf("%d",&m); int q,o,v; memset(eg,M,sizeof(eg)); for(i=1;i<=n;i++) { scanf("%d%d%d",&q,&o,&v); eg[o][q]=eg[q][o]=v; } memset(vid,0,sizeof(vid)); prim(); } return 0; }