Ruby でarrayの中身のint を比較する

Question

ゲームアプリでよくある「XXつのポイントを獲得しました！」のように、一定のポイント数になったらテキストを表示するような処理をシンプルに書く方法がないでしょうか？

具体的には、

point modelの中のクラスメソッドで、受け取った point が以下の

POINT_STEP = [5,10,15,20,25,30,50,100]

というステップのどこにいるか、を処理して、達成しているステップがあれば、そのintを抜き出して、 pointテーブルのstepカラムにそのintを入れるようなイメージです。

例えば、
point = 3 > (達成したステップがないので) step: null
point = 5 > (POINT_STEPの5を達成したので) step: 5
point = 12 > (POINT_STEPの10を達成したので) step: 10

というかたちをシンプルに書けないでしょうか。

Answer 1

POINT_STEP.sort.reverse.bsearch(&point.method(:>=))

※ (&point.method(:>=))部分は{ |x| point >= x }と同じです。

とすれば実現出来るかと思います。今回のように既にソート済みならsortは書く必要はありません。

単純に「ある条件を満たす最初の要素」を返す場合はfindが使えます。しかし、大きさを比較可能な要素からなるソート済みの配列の場合はbsearhを使った方が高速になります。ただここで問題なのは与えられたPOINT_STEPは昇順であるため、『ある値以下の最後の要素』を探す必要があり、これでは「ある条件を満たす最初の要素」ではなくという「ある条件を満たす最後の要素」の形になってしまいます。それを実現するようなメソッドはありませんが、逆に降順であったならば、『ある値以下の最初の要素』を探すことになり、「ある条件を満たす最初の要素」の形になります。つまり、

ある数列において有る値以下の数のうちで最も大きい数

昇順で並んだ数列のある値以下の最後の数

降順で並んだ数列のある値以下の最初の数

は同じであると言うことです。このことから、POINT_STEPを昇順でソート(sort)し、逆順にする(reverse)することで降順の数列を得て、その数列において、最初に現れるpoint以下の数を求める(bsearch(&point.method(:>=)))とすればよいとなります。

Answer 2

POINT_STEPS = [5,10,15,20,25,30,50,100]

def point_to_step(point)
  POINT_STEPS.reverse_each do |p|
    return p if p < point
  end
  nil
end

[0,3, 10, 99, 101].each do |p|
  puts "#{p} -> #{point_to_step(p)}"
end

のような point_to_step メソッドをつくって

...
step: point_to_step(point),
...

とすればよいのでは？

Answer 3

2分探索する場合

POINT_STEP = [5,10,15,20,25,30,50,100]

def search(i)
  ix = POINT_STEP.bsearch_index{|x| x > i} || POINT_STEP.size
  POINT_STEP[ix - 1] if ix > 0
end

p [3, 5, 12, 2500].map{|i| search(i)}
# => [nil, 5, 10, 100]

線形探索する場合

POINT_STEP = [5,10,15,20,25,30,50,100]

def search(i)
  ix = POINT_STEP.rindex{|x| x <= i}
  POINT_STEP[ix] if ix
end

p [3, 5, 12, 2500].map{|i| search(i)}
# => [nil, 5, 10, 100]

Comments

[asm]

rindexよりもtake_while{|x| x <= i}.lastのがよさげ