PKU 1251 最小生成树

背景：

无向连通图G=(V,E)的最小生成树算法：

1.prim算法：

    (1)初始顶点集合S为空,把任意一个顶点s加入集合S.

    (2)迭代|V|-1次，每次加入顶点u，使得v属于S，u不属于S并且(u,v)权值最小, 记录边(u,v)

    (3)所有记录过的边(u,v)即最小生成树

2.kruskal算法：

    (1)把有|E|中边按权值非降排序

    (2)边初始集合S为空，每次加入权值最小的边，当加入后边集合不构成回路，删掉此边。回路判定可以采取并查集方法

    (3)边集合为最小生成树

 

题意：求n个村庄的最小生成树，1<n<27,n个村庄用前n个大写字母表示。输入n和 n-1个顶点与之连通的顶点和权值。输入最小生成树边权值和。边权值小于100，边集合|E|小于等于75，一个顶点连接的边数小于等于15.

分析：prim算法对于处理稠密邻接矩阵有效，而kruskal对于边稀疏矩阵有效。这道题应该kruskal更有效一些。不过两个算法在Judge上运行都是Memory 228K, 时间为0MS，不知道那些Memory<30K是怎么优化的。

 

源码：

 #include <iostream> using namespace std; const int INF = 101; struct node{ node():id(0),dis(0),next(NULL){} node(char i,char d):id(i),dis(d),next(NULL){} char id; char dis; struct node * next; }; /*a为邻接表*/ int kruskal(node *a, int n) { char *set = new char[n];/*并查集*/ char top = 0;/*并查集构造时参数*/ int c, sum, src; node *it, *des, init(0,INF); c = sum = 0; memset(set,0,sizeof(char)*n); set[top++]='A' + top; while(c!=n-1){ des = &init;//初始距离为INF for(int i=0;i<n-1;i++){//迭代邻接表所有边找未用过并且权值最小的边 it = &a[i]; while(it->next!=NULL){ if(des->dis > it->next->dis && it->next->id!=INF){ src = i; des = it->next; } it=it->next; } } if(set[src]!=0 && set[des->id]!=0){//如果两个顶点都在集合中 if(set[src] != set[des->id]){/*无回路*/ for(int i=0;i<n;i++) if(set[i]==set[des->id] && i!=des->id)//合并集合 set[i] = set[src]; set[des->id]=set[src]; des->id=INF;sum +=des->dis;c++;//边置为无效,求最小生成树边权和,边数加一 }else//有回路 des->id=INF; }else if(set[src]==0 && set[des->id]!=0){ set[src]=set[des->id];//加入集合 des->id=INF;sum +=des->dis;c++; }else if(set[src]!=0 && set[des->id]==0){ set[des->id]=set[src];//加入集合 des->id=INF;sum +=des->dis;c++; }else if(!set[src]&&!set[des->id]){ set[src]=set[des->id]='A'+top++;//建立新集合 des->id=INF;sum +=des->dis;c++; } } delete [] set; return sum; } int main() { int num; while(true){ /*建图*/ cin >> num; if(num==0) break; node* adj = new node[num-1]; for(int i=0;i<num-1;i++){ int n, dis; char src, des; cin >> src >> n; for(int j=0;j<n;j++){ cin >> des >> dis; node* tmp=&adj[src-'A']; while(tmp->next!=NULL) tmp = tmp->next; tmp->next = new node(des-'A',dis); } } /*kruskal求最小生成树，adj为邻接表,num为顶点数目*/ cout << kruskal(adj,num) << endl; /*释放申请空间*/ for(int i=0;i<num-1;i++){ node *tmp= adj[i].next,*tmp1; while(tmp!=NULL){ tmp1=tmp; tmp=tmp->next; delete tmp1; } } delete [] adj; } return 0; }

 

//prim算法 #include <iostream> using namespace std; const char N = 26; const char INF = 101; int prim(char *a, int n) { int sum = 0; char i,j,c=n; char min, src,des; bool *is = new bool[n]; memset(is,0,sizeof(bool)*n); is[0]=true; sum = 0; while(--c){ min = INF; for(i=0;i<n;i++) if(true == is[i]) for(j=0;j<n;j++) if(false == is[j]) if(min > a[i*n+j]){ min = a[i*n+j]; src = i; des = j; } is[des]=true; sum += a[src*n+des]; } delete [] is; return sum; } int main() { int t, dis, n; char *adj; char src,des; while(true){ cin >> t; adj = new char[t*t]; memset(adj,INF,t*t*sizeof(char)); if(t==0) break; for(int i=0;i<t-1;i++){ cin >> src >> n; for(int j=0;j<n;j++){ cin >> des >> dis; adj[(src-'A')*t+(des-'A')] = dis; adj[(des-'A')*t+(src-'A')] = dis; } } cout << prim(adj,t) << endl; delete [] adj; } return 0; }