归并集

/********************************** * 归并集 :) * 包括find和union两个操作 * find 定义为查找到data的根 * union_oper 定义为归并两个树 * 其中find的过程会对路径进行压缩 * union的时候通过秩引导降低树高 * * author : huangkq19989 * blog : http://blog.csdn.net/kangquan2008 * * ********************************/ #include <stdio.h> #include <stdlib.h> #define SIZE 5 int parent[SIZE]; int rank[SIZE]; //-1,0,2,3,4,6,7,9,1,-1 //0 1 2 3 4 5 6 7 8 9 int find(int data) { int p = data; // 查找data的根，用p保存 while(parent[p] != -1) p = parent[p]; int q = data; // 压缩 while(parent[q] != -1) { int tmp = q; q = parent[q]; parent[tmp] = p; } return p; } void union_oper(int tree1,int tree2) { int p1 = find(tree1); int p2 = find(tree2); if(p1 != p2) if(rank[p1] < rank[p2]) parent[p1] = p2; else if(rank[p1] > rank[p2]) parent[p2] = p1; else { parent[p1] = p2; rank[p2] ++; } } void print() { for(int i=0; i<SIZE; i++) printf("%d ",parent[i]); printf("\n"); } int main() { for(int i=0; i<SIZE; i++) { parent[i] = -1; rank[i] = 0; } print(); while(1) { printf("input the index of data you want union\n"); int a,b; scanf("%d",&a); scanf("%d",&b); if(a >= SIZE || b >= SIZE) break; union_oper(a,b); print(); } return 0; }