LeetCode(74):搜索二维矩阵

时间:2022-06-17 07:26:15

Medium!

题目描述:

编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:

  • 每行中的整数从左到右按升序排列。
  • 每行的第一个整数大于前一行的最后一个整数。

示例 1:

输入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 3
输出: true

示例 2:

输入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 13
输出: false

解题思路:

这道题要求搜索一个二维矩阵,由于给的矩阵是有序的,所以很自然的想到要用二分查找法,我们可以在第一列上先用一次二分查找法找到目标值所在的行的位置,然后在该行上再用一次二分查找法来找是否存在目标值。

C++解法一:

 // Two binary search
class Solution {
public:
bool searchMatrix(vector<vector<int> > &matrix, int target) {
if (matrix.empty() || matrix[].empty()) return false;
if (target < matrix[][] || target > matrix.back().back()) return false;
int left = , right = matrix.size() - ;
while (left <= right) {
int mid = (left + right) / ;
if (matrix[mid][] == target) return true;
else if (matrix[mid][] < target) left = mid + ;
else right = mid - ;
}
int tmp = right;
left = ;
right = matrix[tmp].size() - ;
while (left <= right) {
int mid = (left + right) / ;
if (matrix[tmp][mid] == target) return true;
else if (matrix[tmp][mid] < target) left = mid + ;
else right = mid - ;
}
return false;
}
};

这道题也可以使用一次二分查找法,如果我们按S型遍历该二维数组,可以得到一个有序的一维数组,那么我们只需要用一次二分查找法,而关键就在于坐标的转换,如何把二维坐标和一维坐标转换是关键点,把一个长度为n的一维数组转化为m*n的二维数组(m*n = n)后,那么原一维数组中下标为i的元素将出现在二维数组中的[i/n][i%n]的位置,有了这一点,代码很好写出来了。

C++解法二:

 // One binary search
class Solution {
public:
bool searchMatrix(vector<vector<int> > &matrix, int target) {
if (matrix.empty() || matrix[].empty()) return false;
if (target < matrix[][] || target > matrix.back().back()) return false;
int m = matrix.size(), n = matrix[].size();
int left = , right = m * n - ;
while (left <= right) {
int mid = (left + right) / ;
if (matrix[mid / n][mid % n] == target) return true;
else if (matrix[mid / n][mid % n] < target) left = mid + ;
else right = mid - ;
}
return false;
}
};