C++数据结构等价类实现

struct cell { int father; int count; }; const int maxlength= 20; void Union(int a, int b, cell* c); int find(int a, cell* c); void initial(int a, cell* c); void equiva(cell* c, int n); void output(cell* c, int n); int main() { cell c[maxlength]; int i, m; cout<<"输入集合元素，以-1结束\n"; for(i=0; m>0; i++) //输入开始的各个元素 { cin>>m; if(m>0) initial(m, c); //把每个元素建立一个自己的集合 } i--; equiva(c, i); //求出数组c中的等价类 cout<<"输出各等价类:\n"; output(c, i); //输出等价类 return 0; } void Union(int a, int b, cell* c) //合并数组c中的a， b两个集合 { if(c[a].count>c[b].count) { c[b].father=a; c[a].count+=c[b].count; } else { c[a].father=b; c[b].count+=c[a].count; } } int find(int a, cell* c) //在数组c中查找元素a所在的集合的根结点 { int i=a; while(c[i].father) i=c[i].father; return i; } void initial(int a, cell* c) //把数组c中的每个元素各建立一个只包含元素自己集合 { c[a].father=0; c[a].count=1; } void equiva(cell* c, int n) //n记录c数组中元素的个数，求出数组c中的等价类 { int i, j, m, k; for(i=1; i<=n; i++) { initial(i, c); } cout<<"输入等价的结点，以（0，0）结束\n"; cin>>i>>j; //输入等价的结点，以（0，0）结束 while((i||j)) { k=find(i,c); m=find(j,c); if(k != m) Union(k, m, c); cin>>i>>j; } } void output(cell* c, int n) //输出数组c中的等价类， n为元素的个数 { int i, j=0, b[maxlength], m; for(i=1; i<=n; i++) { if(c[i].father==0) { b[j++]=i; } } m=j; for(i=0; i<m; i++) { for(j=1; j<=n; j++) { if(b[i]==find(j, c)) cout<<j<<" "; } cout<<endl; } }