質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

ただいまの
回答率

88.93%

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

解決済

回答 3

投稿

  • 評価
  • クリップ 0
  • VIEW 166

torisan3

score 35

やりたい事は、配列(要素数4固定)の値の中で最大値の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


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

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

  • 気になる質問をクリップする

    クリップした質問は、後からいつでもマイページで確認できます。

    またクリップした質問に回答があった際、通知やメールを受け取ることができます。

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 過去に投稿した質問と同じ内容の質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

回答 3

checkベストアンサー

+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

投稿

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2020/07/18 11:34

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

    キャンセル

+1

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

投稿

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2020/07/18 11:39

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

    キャンセル

+1

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]

投稿

編集

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2020/07/18 11:35

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

    キャンセル

15分調べてもわからないことは、teratailで質問しよう!

  • ただいまの回答率 88.93%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる

関連した質問

同じタグがついた質問を見る