LeetCode 搜索二维矩阵 / search a 2D matrix

最新推荐文章于 2023-06-08 09:41:16 发布

twt727

最新推荐文章于 2023-06-08 09:41:16 发布

阅读量256

点赞数

分类专栏：刷题文章标签： leetcode onlineJudge

本文链接：https://blog.csdn.net/twt727/article/details/80102140

版权

刷题专栏收录该内容

26 篇文章 0 订阅

订阅专栏

本文介绍了如何解决LeetCode中的二维矩阵搜索问题，通过双重二分查找法，重点解析了边界判断及其优化策略，确保算法高效运行。并提供了详细的解题代码。

摘要由CSDN通过智能技术生成

题目描述：

解题思路：题目要求尽可能快，自然想到双重二分，难点是边界判断要搞清楚，建议写之前大脑先模拟一下，头尾两个指针（start 和 end）越考越近的时候会发生什么。

这时候会出现两种情况（1）：start 和 end 重合，然后在循环判断一次结束

（2）：start 和 end 相邻，然后在循环判断一次，这时候可能直接结束也有可能转换成情况（1）

模拟一下你就会发现无论是以哪种情况结束，答案都会在 matrix[mid] 或 matrix[mid - 1] 这两行中产生，所以第二次二分时直接对这两行判断即可。代码如下：

class Solution {
	public:
		bool searchMatrix(vector<vector<int>>& matrix, int target) {
			int m = matrix.size();
            if(m == 0) return false;
			int n = matrix[0].size();
            if(n == 0) return false;
			int start = 0,end = m - 1;
			int start0,end0,mid,mid0;
			while(start <= end){    //第一次二分
				mid = (start + end) / 2;
				if(matrix[mid][0] > target){
					end = mid - 1;
				}
				else if(matrix[mid][0] < target){
					start = mid + 1;
				}
				else{
					return true;
				}
			}
			if(matrix[mid][0] < target){  //第二次二分
			//答案可能出现在 matrix[mid] 或 matrix[mid - 1]这两行中的其中一行
				n = matrix[mid].size();
				start0 = 0;
				end0 = n - 1;
				while(start0 <= end0){
					mid0 = (start0 + end0) / 2;
					if(matrix[mid][mid0] > target){
						end0 = mid0 - 1;
					}
					else if(matrix[mid][mid0] < target){
						start0 = mid0 + 1;
					}
					else{
						return true;
					}
				}
			}
			else if(mid - 1 >= 0 && matrix[mid - 1][0] < target){ //答案可能出现在 matrix[mid] 或 matrix[mid - 1]这两行
				n = matrix[mid - 1].size();
				start0 = 0;
				end0 = n - 1;
				while(start0 <= end0){
					mid0 = (start0 + end0) / 2;
					if(matrix[mid - 1][mid0] > target){
						end0 = mid0 - 1;
					}
					else if(matrix[mid - 1][mid0] < target){
						start0 = mid0 + 1;
					}
					else{
						return true;
					}
				}
			}
			return false;
		}
	};