leetcode 240. 搜索二维矩阵 II

题目描述

编写一个高效的算法来搜索 m x n 矩阵 matrix 中的一个目标值 target。该矩阵具有以下特性:

  • 每行的元素从左到右升序排列。
  • 每列的元素从上到下升序排列。

示例:

现有矩阵 matrix 如下:
[
  [1,   4,  7, 11, 15],
  [2,   5,  8, 12, 19],
  [3,   6,  9, 16, 22],
  [10, 13, 14, 17, 24],
  [18, 21, 23, 26, 30]
]
给定 target = 5,返回 true。
给定 target = 20,返回 false。

题解

因为数组从左到右,从上到下都是递增的,所以可以从数组左下角的开始判断,比目标值大就往上比较,比目标小就往右比较

示例代码(go)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
func searchMatrix(matrix [][]int, target int) bool {
m, n := len(matrix), 0
if m > 0 {
n = len(matrix[0])
}
for i, j := m-1, 0; i >= 0 && j < n; {
if matrix[i][j] > target {
i--
} else if matrix[i][j] < target {
j++
} else {
return true
}
}
return false
}