Leetcode 240 - Search a 2D Matrix II

题意

求杨氏矩阵内数x是否存在。

思路

我们将右上角的数设置为起点。

若$x = a[i,j]: return$

若$x > a[i, j]: i++$

若$x < a[i, j]: j--​$

代码

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int m = matrix.size();
        if (m) {
            int n = matrix[0].size();
            int i = 0, j = n - 1;
            while (1) {
                if (i < 0 || j < 0 || i >= m || j >= n) return 0;
                if (target == matrix[i][j]) return 1;
                if (target > matrix[i][j]) i++;
                else j--;
            }
        }
        return 0;
    }
};