&LeetCode0240&搜索一个二维数组之二

题目

编写一个高效的算法来搜索 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。

来源：力扣（LeetCode）

思路

可以观察到有两个位置的数字很有特点，左下角和右上角的数。左下角的18，往上所有的数变小，往右所有数增加，所以就可以和目标数相比较，如果目标数大，就往右搜，如果目标数小，就往上搜。这样就可以判断目标数是否存在。当然也可以把起始数放在右上角，往左和下搜，停止条件设置正确就行。

C++代码

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) 
    {
        if (matrix.empty() || matrix[0].empty())
            return false;

        if (target < matrix[0][0] || target > matrix.back().back())
            return false;

        int x = matrix.size() - 1, y = 0;
        while(true)
        {
            if (matrix[x][y] > target)
                --x;
            else if (matrix[x][y] < target)
                ++y;
            else
                return true;
            
            if (x < 0 || y >= matrix[0].size())
                break;
        }     
        return false;
    }
};