Contents

# frozen_string_literal: true

class Node
  attr_reader :weight

  def initialize(value, weight, left = nil, right = nil)
    @value = value
    @weight = weight
    @left = left
    @right = right
    @leaf = (@left == nil && @right == nil)
  end

  def traverse(code, hash)
    if @leaf
      hash[@value] = code
    else
      @left.traverse("#{code}1", hash)
      @right.traverse("#{code}0", hash)
    end
  end

  def <=>(other)
    @weight <=> other.weight
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
huffman_coding-1.1.0	lib/huffman_coding/node.rb
huffman_coding-1.0.0	lib/huffman_coding/node.rb