리트코드 240번 Search a 2D Matrix 2 (python)

Problem

https://leetcode.com/problems/search-a-2d-matrix-ii/description/

Solution

내 풀이

import bisect
from typing import List

class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        for i in range(len(matrix)):
            if matrix[i][0] <= target:
                idx = bisect.bisect_left(matrix[i], target)
                if idx < len(matrix[i]) and matrix[i][idx] == target:
                    return True
            else:
                return False
        return False

row의 제일 첫 원소가 target보다 같거나 작을 때 해당 row에서 target이 있는지 bisect를 이용해서 찾았다.

다른 풀이

from typing import List

class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        return any(target in row for row in matrix)

파이썬스러운 방법이다.

Reference

파이썬 알고리즘 인터뷰 69번