邻接数组Prim（POJ1287）

来源：互联网发布：gcc linux 安装编辑：程序博客网时间：2024/06/05 16:02

链接：http://poj.org/problem?id=1287

代码（C++）#include <stdio.h>int adjMatrix[100][100];const int MAXLEN = 1000;bool known[100];int dv[100];void initGraph(int,int); int smallestUnkownVertex(int); int prim(int);void initGraph(int p,int r){for(int i=0;i<p;i++){for(int j=0;j<p;j++)adjMatrix[i][j] = MAXLEN;}for(int i=0;i<r;i++){int a,b,dis;scanf("%d %d %d",&a,&b,&dis);if(dis < adjMatrix[a-1][b-1] && dis < adjMatrix[b-1][a-1])adjMatrix[a-1][b-1] = adjMatrix[b-1][a-1] = dis;}}int smallestUnkownVertex(int n) { int min = MAXLEN; int res = -1; for(int i=0;i<n;i++) { if(dv[i]<=min && !known[i]) { min = dv[i]; res = i; } } return res; }int prim(int n) {int size = n; for(int i=0;i<size;i++) { known[i] = false; dv[i] = MAXLEN; } dv[0] = 0; while(true) { int v = smallestUnkownVertex(size);if(v < 0) break; known[v] = true; for(int i=0;i<size;i++){ if(!known[i]) if(adjMatrix[v][i] < dv[i]) dv[i] = adjMatrix[v][i]; } } int res = 0; for(int i=0;i<size;i++) res += dv[i]; return res; }int main(){int P,R;while(scanf("%d",&P) != EOF && P != 0){scanf("%d",&R);if(R == 0)printf("0/n");else{initGraph(P,R);printf("%d/n",prim(P));}}}

邻接数组比邻接表编程容易多了……