# -*- coding: utf-8 -*-
# q1010_answer_1.rb
 
# 工夫した点・苦労した点・感想等（ご自由にお書きください）：
# ・メインとなる iter_a は each メソッドでループを回し、iter_b は Enumerator 化して next（および peek）メソッドを利用するようにしました。
# ・next メソッドが返す値と、返した後で内部で保持している値（＝その後に peek メソッドが返す値）が異なるので、呼ぶ順番に苦労しました。
 
# iter_a に存在し、iter_b に出てこない要素（整数）を列挙するメソッド
def enum_diff iter_a, iter_b
  return to_enum :enum_diff, iter_a, iter_b unless block_given?
  enum_b = iter_b.to_enum
  enum_b_terminated = false
  iter_a.each do |a|
    if enum_b_terminated
      yield a
    else
      begin
        b = enum_b.peek
        while b < a
          enum_b.next
          b = enum_b.peek
        end
        yield a if a < b
      rescue StopIteration
        yield a
        enum_b_terminated = true
      end
    end
  end
end
 
# ※これ以降は変更しないこと。
if $0 == __FILE__
  require 'test/unit'
  require 'prime'
 
  def primes
    Prime.to_enum
  end
 
  def nats
    (1...Float::INFINITY)
  end
 
  def fib
    return to_enum :fib unless block_given?
    a, b = 0, 1
    loop do
      yield b
      a, b = b, a + b
    end
  end
 
  class AnswerQ2Test < Test::Unit::TestCase
    def test_fib
      expected = [1, 1, 2, 3, 5, 8, 13, 21, 34, 55]
      result = enum_diff(fib, []).take(10)
      assert_equal(expected, result)
    end
 
    def test_nats_without_fib
      expected = [4, 6, 7, 9, 10, 11, 12, 14, 15, 16]
      result = enum_diff(nats, fib).take(10)
      assert_equal(expected, result)
    end
 
    def test_fib_without_primes
      expected = [1, 1, 8, 21, 34, 55, 144, 377, 610, 987, 2584, 4181, 6765, 10946, 17711, 46368]
      result = enum_diff(fib, primes).take_while{|n| n < 50000}
      assert_equal(expected, result)
    end
 
    def test_fib_reflect
      # 3以上20以下の整数を乱択
      k = rand(18) + 3
      expected = fib.take(k)
      result = enum_diff(fib, enum_diff(nats, fib)).take(k)
      assert_equal(expected, result)
    end
  end
end