[LeetCode] 240. Search a 2D Matrix II

🔦 문제 링크

🔊 파이썬 알고리즘 인터뷰 책을 참고했습니다.

• 문제

m x m 행렬에서 값을 찾아내는 효율적인 알고리즘을 구현하시오.

✍️ 풀이

---

문제가 왼쪽 -> 오른쪽, 위 -> 아래쪽으로 정렬되어 있으므로 맨 오른쪽 위에서 시작하면 target보다 작을 때와 클 때 모두 반응할 수 있다.

🛠 코드

---

class Solution:
    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        row = 0
        col = len(matrix[0]) - 1
        
        while 0 <= row < len(matrix) and 0 <= col < len(matrix[0]):
            if matrix[row][col] == target:
                return True
            elif matrix[row][col] < target:
                row += 1
            else:
                col -= 1
        return False

---

📝 정리

---

🎈 참고

---

Book 링크