多次元配列の中身を条件指定して選んでそのインデックス番号を取得したい

Question

多次元配列arrの中身を条件を指定して選びそのインデックス番号を取得したいです。条件は
１.配列全ての[1]番目の要素が5よりも大きい
2.上の条件に一致した配列が複数あったら[0]番目の要素が大きい方を選ぶ。要素が同じ値だった場合、[1]番目が小さい方の配列（下記の場合だと[8,6]）を選ぶ
この条件で選んだ配列のインデックス番号を求めたいのですがどのようにすればよいかわからず困っております。何か良い方法はないでしょうか？ご回答頂けましたら幸いです。説明が下手くそで申し訳ありません。

let arr = [[8, 10], [8, 6], [7, 4]]

Answer 1

標準メソッドを組み合わせるとこんな感じでしょうか。

var arr = [[8, 10], [8, 6], [7, 4]]

// [1]が5より大きいものを抜き出したのち、条件にあう要素を見つける
let max: [Int] = arr.filter{$0[1] > 5}.reduce([0,0]) {
    if $0[0] == $1[0] {
        if $0[1] < $1[1] {
            return $0
        } else {
            return $1
        }
    } else if $0[0] < $1[0] {
        return $1
    } else {
        return $0
    }
}
// 条件にあう要素を抽出
print(max)

// 条件にあう要素がある最初のインデックス番号を抽出
print("最初のインデックス番号は: ", arr.firstIndex(of: max)!)

ただし、空配列などのチェックは全く考慮していないので、その辺りは必要に応じて実装が必要かと思います。

詳解Swift第5版　CHAPTER13：ジェネリクスのreduce項あたりを参考にしました。

Comments

[okkyu]

ご回答ありがとうございます。reduceは合計出すことでしかまだ使ったことなかったのでとても参考になりました。