【二分】74. Search a 2D Matrix

原创于 2018-05-08 12:41:51 发布

· 73 阅读

0 ·

版权

LeetCode 专栏收录该内容

107 篇文章

订阅专栏

本文介绍了一种高效的搜索算法，用于在一个有序的二维矩阵中查找特定元素。该算法分为两个步骤：首先通过二分查找确定目标元素可能存在的行，然后在该行内再次使用二分查找精确定位目标元素。文章提供了完整的C++实现代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

第一步：二分查找元素可能所在行；

第二步：在改行查找该元素是否存在

class Solution {
public:
   bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int row_begin=0,row_end=matrix.size()-1;
        if(row_end<0||(matrix[0].size()<1))
            return 0;
        int row_mid=row_begin+(row_end-row_begin)/2;
        while(row_end-row_begin > 1)
        {
        	if(matrix[row_mid][0]==target)
        		return 1;
        	else
        		if(matrix[row_mid][0]<target)
        			row_begin=row_mid;
        		else
        			row_end=row_mid;

        	row_mid=row_begin+(row_end-row_begin)/2;
        }
       if(matrix[row_end][0]<=target)
        	row_begin=row_end;
        else
        	if(matrix[row_begin][0]>target)
        		return 0;


        int begin=0,end=matrix[row_begin].size()-1;
        int mid= begin+(end-begin)/2;
        while(begin<=end)
        {
        	if(matrix[row_begin][mid]==target)
        		return 1;
        	else
        	{
        		if(matrix[row_begin][mid]<target)
        			begin=mid+1;
        		else
        			end=mid-1;
        	}
        	mid= begin+(end-begin)/2;
        }
        return 0;
    }
};