剑指offer04

题目描述

解题思路

该矩阵的特点是左上角的元素最小，右下角的元素最大。
左下角的元素是同一行中最小的，也是同一列中最大的。
我们开始从左下角开始查找，如果目标值比数组元素大，就到右边去寻找，如果目标值比数组元素小，就到上边去寻找。
类似的，我们也可以从右上角开始查找。

代码

class Solution {
public:
    bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
        if (matrix.size() == 0){
            return false;
        }
        int m = matrix.size();
        int n = matrix[0].size();
        int row = m - 1;
        int col = 0;
        while (row >=0 && col < n){
            if(matrix[row][col] < target){
                col += 1;
            }
            else if(matrix[row][col] > target){
                row -= 1;
            }
            else{
                return true;
            }
        }
        return false;
    }
};