240. Search a 2D Matrix II
Содержание
Задача
Вам дана двумерная матрица размера m x n, представляющая прямоугольник, и целое число target
. Отсортированные строки матрицы по неубыванию с лева направо и столбцы отсортированы по неубыванию сверху вниз.
Найдите элемент target
в матрице. Верните True
, если элемент target
есть в матрице, и False
, если его нет.
Подсказки
Попробуйте использовать двоичный поиск для каждого ряда.
Подход
Самый простой подход - использовать двоичный поиск для каждого ряда. Хотя это не самый эффективный метод, он достаточно прост для понимания.
Алгоритм
- Пройдитесь по каждому ряду в матрице.
- В каждом ряду используйте двоичный поиск для поиска
target
.
Решение
from bisect import bisect_left
def searchMatrix(matrix, target):
for row in matrix:
pos = bisect_left(row, target)
if pos != len(row) and row[pos] == target:
return True
return False