POJ2421 prim算法求最小生成树
题意：邻接村子里面所有的村庄，使得道路的总长度最小。
分析：由于题意说会有已经通好道路的村庄。我们任然可以利用prim求解最小生成树，只要把已经通好道路的权值置为0，就行了。
View Code
1 // I'm the Topcoder 2 //C 3 #include <stdio.h> 4 #include <stdlib.h> 5 #include <string.h> 6 #include <ctype.h> 7 #include <math.h> 8 #include <time.h> 9 //C++ 10 #include <iostream> 11 #include <algorithm> 12 #include <cstdio> 13 #include <cstdlib> 14 #include <cmath> 15 #include <cstring> 16 #include <cctype> 17 #include <stack> 18 #include <string> 19 #include <list> 20 #include <queue> 21 #include <map> 22 #include <vector> 23 #include <deque> 24 #include <set> 25 using namespace std; 26 27 //*************************OUTPUT************************* 28 #ifdef WIN32 29 #define INT64 "%I64d" 30 #define UINT64 "%I64u" 31 #else 32 #define INT64 "%lld" 33 #define UINT64 "%llu" 34 #endif 35 36 //**************************CONSTANT*********************** 37 #define INF 0x3f3f3f3f 38 #define eps 1e-8 39 #define PI acos(-1.) 40 #define PI2 asin (1.); 41 typedef long long LL; 42 //typedef __int64 LL; //codeforces 43 typedef unsigned int ui; 44 typedef unsigned long long ui64; 45 #define MP make_pair 46 typedef vector<int> VI; 47 typedef pair<int, int> PII; 48 #define pb push_back 49 #define mp make_pair 50 51 //***************************SENTENCE************************ 52 #define CL(a,b) memset (a, b, sizeof (a)) 53 #define sqr(a,b) sqrt ((double)(a)*(a) + (double)(b)*(b)) 54 #define sqr3(a,b,c) sqrt((double)(a)*(a) + (double)(b)*(b) + (double)(c)*(c)) 55 56 //****************************FUNCTION************************ 57 template <typename T> double DIS(T va, T vb) { return sqr(va.x - vb.x, va.y - vb.y); } 58 template <class T> inline T INTEGER_LEN(T v) { int len = 1; while (v /= 10) ++len; return len; } 59 template <typename T> inline T square(T va, T vb) { return va * va + vb * vb; } 60 61 // aply for the memory of the stack 62 //#pragma comment (linker, "/STACK:1024000000,1024000000") 63 //end 64 65 #define maxn 2000+10 66 int n,m; 67 int edge[maxn][maxn];//邻接矩阵 68 int lowcost[maxn]; 69 int nearvex[maxn]; 70 int sumweight=0; 71 void prim(int u0){ 72 //从顶点u0出发执行普里姆算法 73 sumweight=0;//生成树的权值 74 for(int i=1;i<=n;i++){ 75 //初始化lowcost[]数组和neartxt数组 76 lowcost[i]=edge[u0][i]; 77 nearvex[i]=u0; 78 } 79 nearvex[u0]=-1; 80 for(int i=1;i<n;i++){ 81 int min=INF; 82 int v=-1; 83 //在lowcoat数组的nearvex[]值为-1的元素中找最小值 84 for(int j=1;j<=n;j++){ 85 if(nearvex[j]!=-1&&lowcost[j]<min){ 86 v=j; 87 min=lowcost[j]; 88 } 89 } 90 if(v!=-1){ 91 //v==-1表示没找到权值最小的边 92 // printf("%d %d %d\n",nearvex[v],v,lowcost[v]); 93 nearvex[v]=-1; 94 sumweight+=lowcost[v]; 95 for(int j=1;j<=n;j++){ 96 if(nearvex[j]!=-1&&edge[v][j]<lowcost[j]){ 97 lowcost[j]=edge[v][j]; 98 nearvex[j]=v; 99 } 100 } 101 } 102 } 103 printf("%d\n",sumweight); 104 } 105 106 int main(){ 107 int u,v,w; 108 while(scanf("%d",&n)!=EOF){ 109 sumweight=0; 110 //memset(edge,0,sizeof(edge)); 111 for(int i=1;i<=n;i++){ 112 for(int j=1;j<=n;j++){ 113 edge[i][j]=INF; 114 } 115 edge[i][i]=0; 116 } 117 for(int i=1;i<=n;i++){ 118 for(int j=1;j<=n;j++){ 119 scanf("%d",&w); 120 edge[i][j]=w; 121 } 122 } 123 scanf("%d",&m); 124 for(int i=1;i<=m;i++){ 125 scanf("%d%d",&u,&v); 126 edge[u][v]=edge[v][u]=0; 127 } 128 // for(int i=1;i<=n;i++){ 129 // for(int j=1;j<=n;j++){ 130 // if(i==j) edge[i][j]=0; 131 // else if(edge[i][j]==0 ) edge[i][j]=INF; 132 // } 133 // } 134 prim(1); 135 } 136 return 0; 137 }
