【力扣hot100题】(020)搜索二维矩阵Ⅱ

一开始以为一行的末尾小于下一行的开头,于是傻傻的写了两二分查找,所幸的是发现了自己在二分查找上确实不怎么熟练,巩固了一下知识。

这是一开始写的错误答案:

cpp 复制代码
class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int left=0;
        int right=matrix[0].size()-1;
        int record=0;
        while(left<=right){
            int middle=left+(right-left)/2;
            if(matrix[0][middle]>target) right=middle-1;
            else if(matrix[0][middle]<target) {record=left;left=middle+1;}
            else return 1;
        }
        left=0;
        right=matrix.size()-1;
        while(left+1<right){
            int middle=(left+right+1)/2;
            if(matrix[middle][record]>target) right=middle;
            else if(matrix[middle][record]<target) left=middle;
            else return 1;
        }
        if(matrix[left][record]==target||matrix[right][record]==target) return 1;
        return 0;
    }
};

记起来了二分查找中left和right不能只是等于mid,而是要在mid的基础上加或减,不然会死循环,并且跳出while的条件不能包含left==right,两者相等的情况下还要继续二分。

看答案学到了二分查找的逃课快速实现法:

cpp 复制代码
class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        for(auto& row:matrix){
            auto it=lower_bound(row.begin(),row.end(),target);
            if(it!=row.end()&&*it==target) return 1;
        }
        return 0;
    }
};

lower_bound是一个二分查找的逃课函数,传入始末向量和target,就可以二分搜寻出最接近target的不小于target的元素,如果没有将返回end()。

答案还有一种很天才的思路:z字型查找,从右上开始搜索,如果比target小就删除最上行,比target大就删除最右行,非常高效。

cpp 复制代码
class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int n=matrix.size();
        int m=matrix[0].size();
        int x=0;
        int y=m-1;
        while(x<n&&y>=0){
            if(matrix[x][y]>target) y--;
            else if(matrix[x][y]<target) x++;
            else return 1;
        }
        return 0;
    }
};
相关推荐
董董灿是个攻城狮9 小时前
AI视觉连载8:传统 CV 之边缘检测
算法
AI软著研究员16 小时前
程序员必看:软著不是“面子工程”,是代码的“法律保险”
算法
FunnySaltyFish16 小时前
什么?Compose 把 GapBuffer 换成了 LinkBuffer?
算法·kotlin·android jetpack
颜酱17 小时前
理解二叉树最近公共祖先(LCA):从基础到变种解析
javascript·后端·算法
地平线开发者1 天前
SparseDrive 模型导出与性能优化实战
算法·自动驾驶
董董灿是个攻城狮1 天前
大模型连载2:初步认识 tokenizer 的过程
算法
地平线开发者1 天前
地平线 VP 接口工程实践(一):hbVPRoiResize 接口功能、使用约束与典型问题总结
算法·自动驾驶
罗西的思考1 天前
AI Agent框架探秘:拆解 OpenHands(10)--- Runtime
人工智能·算法·机器学习
HXhlx2 天前
CART决策树基本原理
算法·机器学习
Wect2 天前
LeetCode 210. 课程表 II 题解:Kahn算法+DFS 双解法精讲
前端·算法·typescript