240. Search a 2D Matrix II

题目

240. Search a 2D Matrix II

解题思路

这道题与74. Search a 2D Matrix的不同在于这道题目的矩阵是左上角的数字一定比右下角的大。

这道题我是看discussion的题解做的，先把初始位置定在右上角，这样向下移就是增大，向左移就是减小，即有增大减小两个不同方向。

如果把初始位置定在左上角，向右移和向下移都是增大，即两个方向都是增大，无法确定是要哪个方向才能找到target。

具体代码

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int m = matrix.size();
        int n = m > 0 ? matrix[0].size() : 0;
        if (n <= 0) {
            return false;
        }
        // 把初始位置定在右上角
        int i = 0, j = n - 1;
        while (i < m && j >= 0) {
            if (matrix[i][j] == target) {
                return true;
            }
            else if (matrix[i][j] < target) {
                i++; // 下移增大
            }
            else {
                j--;// 左移减小
            }
        }
        return false;
    }
};