プログラマ専用SNS ミクプラ

コード:

#define MAX 3
int p[MAX];
/* union-findの初期化 */
void union_init(void)
{
  int i;
  for(i=0;i<MAX;i++){
    p[i] = i; /* グループが自分のみの状態にする */
  }
}
/* aのリーダーを返す */
int root(int a)
{
  /* グループのリーダーの場合 */
  if(p[a] == a) return a; /* リーダーの番号を返す */
  return p[a]=root(p[a]); /* どこのグループに属しているか分かれば十分なのでリーダーを書き換える */
}
/* aとbが同じグループか */
int same(int a,int b)
{
  return root(a)==root(b);
}
/* aとbを同じグループにする */
void merge(int a,int b)
{
  /* aのリーダーをbのリーダーにする */
  p[root(a)]=root(b);
}