【力扣hot100题】（020）搜索二维矩阵Ⅱ

梭七y2025-04-02 18:57

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

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

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;
    }
};