剑指offer笔记 04.二维数组的查找

二维数组的查找

题目描述

在一个 n * m 的二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个高效的函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。
示例：
[
[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）
链接：https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

C++代码

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

执行结果