Задача: 1380. Lucky Numbers in a Matrix
Сложность: easy

Дана матрица m x n из различных чисел, верните все счастливые числа в матрице в любом порядке.

Счастливое число — это элемент матрицы, который является минимальным элементом в своей строке и максимальным в своем столбце.

Пример:

Input: matrix = [[3,7,8],[9,11,13],[15,16,17]]
Output: [15]
Explanation: 15 is the only lucky number since it is the minimum in its row and the maximum in its column.

👨‍💻 Алгоритм:

1⃣Сохраните минимум каждой строки в список rowMin и максимум каждого столбца в список colMax.

2⃣Итерируйте по каждому числу в матрице и проверяйте, равно ли оно rowMin[i] и colMax[j].

3⃣Если число удовлетворяет условию, добавьте его в список luckyNumbers и верните luckyNumbers.

😎 Решение:

class Solution {
    func luckyNumbers (_ matrix: [[Int]]) -> [Int] {
        let N = matrix.count
        let M = matrix[0].count
        
        var rowMin = [Int]()
        for i in 0..<N {
            var rMin = Int.max
            for j in 0..<M {
                rMin = min(rMin, matrix[i][j])
            }
            rowMin.append(rMin)
        }
        
        var colMax = [Int]()
        for i in 0..<M {
            var cMax = Int.min
            for j in 0..<N {
                cMax = max(cMax, matrix[j][i])
            }
            colMax.append(cMax)
        }
        
        var luckyNumbers = [Int]()
        for i in 0..<N {
            for j in 0..<M {
                if matrix[i][j] == rowMin[i] && matrix[i][j] == colMax[j] {
                    luckyNumbers.append(matrix[i][j])
                }
            }
        }
        
        return luckyNumbers
    }
}

Ставь 👍 и забирай 📚 Базу знаний