牛客JZ1 二维数组中的查找


在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。

[1,2,8,9],
[2,4,9,12],
[4,7,10,13],
[6,8,11,15]

给定 target = 7,返回 true。

给定 target = 3,返回 false。


思路:因为每一行和每一列都是从上到下递增的顺序,所以可以从左下角或者右上角出发搜索,每次可以排除一行或一列。

class Solution {
public:
    bool Find(int target, vector > array) {
        // 从左下角开始搜索
        int row = array.size() - 1, col = 0;
        while (row >= 0 && col < array[0].size()) {
            // 小于target列向右移;大于target则行向上移
            if (array[row][col] < target) {
                col++;
            }
            else if (array[row][col] > target) {
                row--;
            }
            else {
                return true;
            }
        }
        return false;
    }
};