【Leetcode】剑指Offer 04. 二维数组中的查找：利用二分快速查找有序数列

最新推荐文章于 2022-07-17 22:19:53 发布

原创最新推荐文章于 2022-07-17 22:19:53 发布 · 192 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #数据结构 #leetcode #面试

算法模板笔记专栏收录该内容

17 篇文章

订阅专栏

这篇博客探讨了如何利用二分查找算法优化在有序二维数组中搜索目标值的过程。通过逐行应用二分查找，可以显著减少比较次数，提高搜索效率。博主详细解释了算法的实现逻辑，并给出了具体的C++代码示例。

看到这道的第一时间就想到了二分，
因为：

数据的存储形式是数组，可以支持随机访问
数据有序排列，有内部规律

所以我们利用二分对这个过程进行优化
在这里插入图片描述

class Solution {
public:
    bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
        // 这道题肯定不可暴力遍历，看到有序查找，第一时间用二分。
        // 大致思路是：单行的查找用二分加速
        // 遍历是肯定要遍历的，但是要利用上数组的特点进行加速，比如我们要寻找10，那么第一行定位到下标为3，
        //且不为target，那么我们可以确定的是，目标值一定不在第一行第三列右边的区域
        if (matrix.empty() || matrix[0].empty()) return false;
        int left = 0, right = matrix[0].size() - 1;
        for (int i = 0; i < matrix.size(); i++) {
            int mid = 0;
            if (matrix[i][left] == target || matrix[i][right] == target) return true;
            while(left < right) {
                mid = (left + right + 1) >> 1;
                if (matrix[i][mid] == target) return true;
                if (matrix[i][mid] > target) right = mid - 1;
                else left = mid;
            }
            if(matrix[i][mid] > target && (right = mid - 1) < 0) return false;
            right = mid; left = 0;
        }
        return false;
    }
};