配列の中身が条件にあうかどうか

Question

Ruby1.9.4です。

＜前提＞
要素の数が０～４個の配列があります。
値は1～7,11～17,21～27のいずれかが入り、同じ数が重複する場合もあります。
並びは小さい順にソートされています。

＜やりたい事＞
配列の中身の内、いずれか３つの値を調べて
『十の位と一の位がそれぞれ、連続する数値の組み合わせ(順不同)であった時 フラグを立てる。』
という処理を行いたいです。

わかりにくいと思うので例を列挙します。

[]=>false
[1,12]=>false
[2,13,24]=>true #十の位と一の位がそれぞれ、連続する数値である
[4,12,23]=>true #true。一の位が4,2,3だが数値の順序は問わない
[5,6,7]=>false #十の位が一致してしまっている
[7,11,22]=>false #7と1は繋がったりはしない
[3,7,11,22]=>true #余剰データがあってもいずれか３つが条件にあっていればOK
[5,13,13,24]=>true #データが重複していてもOK
[1,4,13,22]=>true #条件が合う組み合わせが２つあっても、ただのtrue

一応自前の処理は動いてはいるのですが、
あまりにも『……。』な処理だったので今回投稿した次第です。
軽い処理が望ましいですが、考え方のみでも結構ですので
お知恵を拝借頂ければと思います。
よろしくお願いします。

Answer 1

自分の回答を置いておきます。

def check0(tmp)
  tmp2=[]
    if tmp.size >= 3 then
      if tmp.size == 4 then #[1,14,17,21] を [1,14,21] と [1,17,21] に分ける
        for i in 0..2
          if tmp[i] / 10 == tmp[i + 1] / 10 then
            tmp2 = Marshal.load(Marshal.dump(tmp))
            tmp.delete_at(i + 1)
            tmp2.delete_at(i)
            break
          end
        end
      end
      tmp3 = [tmp[0] % 10,tmp[1] % 10,tmp[2] % 10].sort
      if tmp3[0] + 1  == tmp3[1] and tmp3[1] + 1 == tmp3[2] then
        tmp3 = [tmp[0] / 10,tmp[1] / 10,tmp[2] / 10].sort
        if tmp3[0] == 0 and tmp3[1] == 1 and tmp3[2] == 2 then
            return true
        end
      end
      if tmp2.size >= 3 then
        tmp3 = [tmp2[0] % 10,tmp2[1] % 10,tmp2[2] % 10].sort
        if tmp3[0] + 1  == tmp3[1] and tmp3[1] + 1 == tmp3[2] then
          tmp3 = [tmp2[0] / 10,tmp2[1] / 10,tmp2[2] / 10].sort
          if tmp3[0] == 0 and tmp3[1] == 1 and tmp3[2] == 2 then
              return true
          end
        end
      end
    end
    return false
end

Answer 2

質問文にある配列の判定はできることを確認しています。

# ary が 連続した３つの数字列をもっているかをチェックする
def has_seq?(ary)
  ary.each_cons(3) do |nums|
    return true if nums[-1] - nums[0] == 2
  end
  false
end

def check(ary)
  keta1 = ary.map { |x| x % 10 }.sort.uniq  # 1の位の数字だけを取り出す
  keta10 = ary.map { |x| x / 10 }.sort.uniq # 10の位の数字だけを取り出す

  return false unless has_seq?(keta1)
  return false unless has_seq?(keta10)
  true
end

TESTS = [
  [[], false],
  [[1, 12], false],
  [[2, 13, 24], true],    # 十の位と一の位がそれぞれ、連続する数値である
  [[4, 12, 23], true],    # true。一の位が4,2,3だが数値の順序は問わない
  [[5, 6, 7], false],     # 十の位が一致してしまっている
  [[7, 11, 22], false],   # 7と1は繋がったりはしない
  [[3, 7, 11, 22], true],  # 余剰データがあってもいずれか３つが条件にあっていればOK
  [[5, 13, 13, 24], true], # データが重複していてもOK
  [[1, 4, 13, 22], true]   # 条件が合う組み合わせが２つあっても、ただのtrue
].freeze

TESTS.each do |test|
  puts "ERROR #{test}" if test[1] != check(test[0])
end

Comments

[torisan]

シンプルでいい感じですね。
回答ありがとうございました。

Answer 3

def check(list)
  list.
    # 無駄な重複を無くします。
    uniq.
    # 10の位と1の位に分離して配列にします。
    map { |i| [i / 10, i % 10] }.
    # 3個を抜き出した全ての組み合わせを羅列します。
    combination(3).
    # 10の位と1の位それぞれをまとめた配列にします。
    map { |a| 2.times.map { |x| 3.times.map { |y| a[y][x] } } }.
    # 組み合わせのうち何れかが条件を満たせばtrueにします。
    any? do |a|
      # 10の位と1の位それぞれについて、両方条件を満たせばtrueにします。
      a.all? do |ak|
        # ソートした後に重複しながら2つ取り出してその差分がすべて1であるかを
        # 確認することで、連続している数値なのかを確認します。
        ak.sort.each_cons(2).all? { |ai| ai[1] - ai[0] == 1 }
      end
    end
end

要素数が多く、組み合わせが膨大になる場合は、combination(3).の後にlazy.を足すと、配列を作らずにtrueが見つかり次第返るようになり、メモリ消費量も抑えられます。(ただし、lazyが使えるのは2.0.0からです)
lazyを使わない場合は、any?前のmapでやっていることをany?の中ですると良いでしょう。そこは自分で工夫してみてください。

Comments

[torisan]

関数をフル活用している感じですね。
条件の指定によっては効率の良いものができるかもしれません。
回答ありがとうございました。