[Ruby] Palindrome, dictionary, anagram

edX ESaaS 수업 첫번째 과제중 하나. word에 대한 palindrome과 단어 반복횟수 체크, 그리고 들어있는 단어에 대한 anagram까지 뽑는게 주제였다.

다른건 쉬웠는데 마지막에 word마다 anagram을 뽑아서 어떻게 처리해야 될지가 약간 고민을 하게 했다.

#!/usr/bin/env ruby

class String
  def palindrome?
    original = self.delete("^a-zA-Z")

    if original.reverse.downcase == original.downcase
      return true
    else
      return false
    end
  end

  def count_words
    original = self.delete("^a-zA-Z ").downcase
    dic = original.split(' ')
    result = Hash.new
    
    dic.each do |element|
      result[element] = 0 
    end

    dic.each do |element|
      if result.has_key?(element)
        result[element] += 1
      end
    end
    return result
  end

  def anagram_groups
    length = self.size
    if length == 0
      return []
    else
      words = self.split(' ')
      dict = Hash.new
      words.each do |word|
        key = word.downcase.chars.sort.join
        if dict.has_key?(key)
          dict[key].push(word)
        else
          dict[key] = [].push(word)
        end
      end
      return dict.values
    end
  end
end