もう少しスマートに書きたいです(配列の最大値のindexを返す)

Question

やりたい事は、配列(要素数４固定)の値の中で最大値のindex値を取得したいです。

ha1 = [200, 250, 350 ,220]
p ha1.index(ha1.max) #=>2

通常時ならこれで事足りるのですが、仕様上最大値を持っているindexが複数ある場合があります。
複数ある場合は、もう一つの配列ha2(ha1と同要素数:4)を見て最小値を持つindexを返したいです。
（※ha2の配列は同一値が入ることはありません。すべてユニークです。）

#ha1で最大なのは420なので3を返す。ha2は無視。
ha1 = [200, 250, 350 ,420]
ha2 = [1, 4, 3 ,2]
--------------------------------------
#ha1で最大なのは100(indexは0と2)。
#ha2[0](1)とha2[2](3)を比較するとha2[0]の方が小さいので0を返す。
ha1 = [100, 20, 100 ,0]
ha2 = [1, 2, 3 ,4]
--------------------------------------
#ha1の値がすべて同じ場合もあり得る。
#ha2の中で1が一番小さいのでindexは2を返す。
ha1 = [200, 200, 200 ,200]
ha2 = [2, 4, 1 ,3]

現状のプログラム

    ha1 = [100, 20, 100 ,100]
    ha2 = [3, 2, 1 ,4]
    tmp = []
    for j in 0..3
      tmp.push(j) if ha1[j] == ha1.max
    end

    ind = tmp[0]
    for j in 1..tmp.size - 1
      ind = tmp[j] if ha2[ind] > ha2[tmp[j]]
    end
    p ind #=>2

値がユニークならたった１行で済む事なので、もう少しスマートに書けないかと思っています。

急いではいないので作業の息抜き時にでも回答していただければと思います。
よろしくお願いいたします。

Answer 1

Enumerable#each_with_index が便利です

hal = [100, 200, 50, 150]
p hal.each_with_index.max  # => [200, 1]

ha1 = [100, 20, 100 ,0]
ha2 = [1, 2, 3 ,4]

p ha1.zip(ha2.map(&:-@)).each_with_index.max[1]  # => 0

ha1 = [100, 20, 100 ,0]
ha2 = [3, 2, 1 ,4]

p ha1.zip(ha2.map(&:-@)).each_with_index.max[1]  # => 2

Comments

[torisan3]

each_with_indexは存在を知りませんでした。
index用の配列を作成しなくてよいのがスマートだと思いました。
回答ありがとうございました！

Answer 2

ha1.zip(ha2.map(&:-@), ha1.size.times).max[2]

これでどうでしょう。
ha1.zip(ha2, ha1.size.times)は[[ha1[0], ha2[0], 0], [ha1[1], ha2[1], 1], ...]というような配列を生成しています。
それをmaxすると、Array#<=>は要素を前から順に比較していく（辞書式）ので、ha1を優先し、ha1が同じならha2を比較し、それが同じならインデックスが大きい方をとる、というふうにしています。
普通のindexはインデックスが小さい方をとるので、そうしたければ↓のようになります。

ha1.size - ha1.zip(ha2.map(&:-@), ha1.size.downto(1)).max[2]

Comments

[torisan3]

ha2.map(&:-@)に最初戸惑ったのですが、ようするに値の正負を入れ替えるということですね。
maxの比較判定方法は知りませんでした。
回答ありがとうございました！

Answer 3

ind = (0...ha1.size).zip(ha1,ha2).max_by{|i,x,y| [x,-y]}[0]

Comments

[torisan3]

max_byが二次元配列にも使えるのは知りませんでした。
書き方も、命令を知らなくてもなんとなく想像できるのがいいですね。
回答ありがとうございました！