ruby 配列のある要素の、隣り合わせの要素を抽出したい

Question

配列のある要素の、隣り合わせの要素のいずれかを抽出したいです。

例えば、次のような条件なときに、
a = 300
list = [50, 100, 200, 300, 500, 1000]

aが300なので、その隣り合わせの”200”または”500”を出力したいです。
（”200”または”500”はランダムで決まる）

ただし、
a = 50やa =1000の時は、それぞれが端なので、
a = 50 -> "100" が出力
a = 1000 -> "500"　が出力
となるようにしたいです。

配列の長さはその都度変わります。
サンプルコードなどを教えて頂けますと大変助かります。

宜しくお願いいたします。

Answer 1

list = [50, 100, 200, 300, 500, 1000]

def find_neighbor(list, a)
  con2 = list.each_cons(2)
  con2.find_all{|x,y| x == a}.map{|x,y| y} + con2.find_all{|x,y| y == a}.map{|x,y| x}
end
p find_neighbor(list, 300) # => [200,500]
p find_neighbor(list, 50) # => [100]
p find_neighbor(list, 1000) # => [500]

Answer 2

こんにちは。
ちょっとベタなコードですが、こんな感じでいかがでしょうか？

def random_neighboring_element(ary, element)
  idx = ary.index(element)
  return unless idx
  neighboring_indexes = [idx - 1, idx + 1].select do |i|
    (0...ary.length).include?(i)
  end
  ary[neighboring_indexes.sample]
end

a = 300
list = [50, 100, 200, 300, 500, 1000]

puts(random_neighboring_element(list, a)) # => 200 or 500

Answer 3

もっとスマートな方法があるかも知れませんが

position = list.index(a)

if position == list.size - 1 || 0
  p (list[position-1] || list[position+1])
else
  p [list[position-1], list[position+1]].sample
end

Comments

[退会済みユーザー]

ありがとうございます。
コピペで試してみたところ上手くいかず、必ず一個左の要素が結果としてでてきます。
a = 300なら必ず200が、a = 50なら必ず1000になってしまいます。
やりたいこととしては、
a = 300なら、200または500のいずれかがランダムに、a = 50なら必ず100が出力されるようにしたいです。

[退会済みユーザー]

教えて頂いたコードをもとに次の様にしたら無事にできました。
ありがとうございました！大変助かりました。
if position == 0
p list[position+1]
elsif position == list.size - 1
p list[position-1]
else
p [list[position-1], list[position+1]].sample
end