题目

在一个 n * m 的二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个高效的函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

示例:
现有矩阵 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。

题解

由于矩阵，从左到右递增，从上到下递增，如果从左上角m[0][0]开始查询，当target > m[0][0]时，无法决定下一步是向右还是向下。
所以，可以选择从左下角m[h][0]开始查询

• 当target > m[h][0]时，向右比较下一个元素m[h][1]
• 当target < m[h][0]时，向上比较下一个元素m[h - 1][0]

代码

func findNumberIn2DArray(matrix [][]int, target int) bool {
	height := len(matrix)
	if height == 0 {
		return false
	}
	width := len(matrix[0])
	h := 0
	for h != height && width > 0 {
		if matrix[h][width-1] > target {
			width -= 1
		} else if matrix[h][width-1] < target {
			h += 1
		} else {
			return true
		}
	}
	return false
}