[剑指offer] 二维数组中的查找

题目描述

在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

---

主要思路是对于数组的每一行进行二分查找。

优化：若某一行的结尾小于目标值或开头大于目标值，则此行一定没有此值直接跳过。

我的代码：

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int len1 = array.size(), len2 = array[0].size();
        if (!len1 || !len2) return false;
        int i = 0, j = len1 - 1;
        while (i < len1 - 1 && array[i][len2-1] < target) i++;  // 注意这里的边界条件
        while (j > 0 && array[j][0] > target) j--;
        if (array[i][len2-1] == target || array[j][0] == target) return true;
        for (; i <= j; i++) {
            int l = 0, r = len2 - 1;
            int mid = (l + r) / 2;
            while (l <= r) {
                mid = (l + r) / 2;
                if (mid < 0 || mid >= len2) break;
                if (array[i][mid] == target) return true;
                if (array[i][mid] > target) r = mid - 1;
                else l = mid + 1;
            }
        }
        return false;
    }
};

题目描述

在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。