Ruby implementation of the quick-union algorithm

quick_union.rb

class QuickUnion

  def initialize(n)
    @id = Array.new(n) {|i| i}
  end

  def root(i)
    while i != @id[i] do
      i = @id[i]
    end
    i
  end

  def connected?(p, q)
    root(p) == root(q)
  end

  def union(p, q)
    @id[p] = root(@id[q])
  end

  def print_union
    puts "#{@id}"
  end

end