LeetCode 240 搜索二维矩阵 II
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-a-2d-matrix-ii/
博主Github:https://github.com/GDUT-Rp/LeetCode
题目:
编写一个高效的算法来搜索 m x n
矩阵 matrix 中的一个目标值 target 。该矩阵具有以下特性:
每行的元素从左到右升序排列。
每列的元素从上到下升序排列。
示例 1:
输入:matrix = [[1,4,7,11,15],[2,5,8,12,19],[3,6,9,16,22],[10,13,14,17,24],[18,21,23,26,30]], target = 5
输出:true
示例 2:
输入:matrix = [[1,4,7,11,15],[2,5,8,12,19],[3,6,9,16,22],[10,13,14,17,24],[18,21,23,26,30]], target = 20
输出:false
提示:
- m == matrix.length
- n == matrix[i].length
- 1 <= n, m <= 300
- − 1 0 9 -10^9 −109 <= matrix[i][j] <= 1 0 9 10^9 109
- 每行的所有元素从左到右升序排列
- 每列的所有元素从上到下升序排列
- − 1 0 9 -10^9 −109 <= target <= 1 0 9 10^9 109
解题思路:
方法一:二分查找
直观想法
map统计各个数的分布情况,然后计算判断是否符合条件。
C++
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
for (const auto& row: matrix) {
auto it = lower_bound(row.begin(), row.end(), target);
if (it != row.end() && *it == target) {
return true;
}
}
return false;
}
};
Golang
func searchMatrix(matrix [][]int, target int) bool {
for _, row := range matrix {
i := sort.SearchInts(row, target)
if i < len(row) && row[i] == target {
return true
}
}
return false
}
Python
class Solution:
def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
for row in matrix:
idx = bisect.bisect_left(row, target)
if idx < len(row) and row[idx] == target:
return True
return False
复杂度分析
时间复杂度:
O
(
m
log
n
)
O(m\log n)
O(mlogn)
空间复杂度:
O
(
1
)
O(1)
O(1)
方法二:左下角开始搜索
从左下脚开始,如果这个数比target小,那就可以往右搜索,否则则往上搜索。
C++
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
int m = matrix.size(), n = matrix[0].size();
int x = 0, y = n - 1;
while (x < m && y >= 0) {
if (matrix[x][y] == target) {
return true;
}
if (matrix[x][y] > target) {
--y;
}
else {
++x;
}
}
return false;
}
};
Go
func searchMatrix(matrix [][]int, target int) bool {
m := len(matrix)
n := len(matrix[0])
i := m - 1
j := 0
for i >= 0 && j < n {
if matrix[i][j] == target {
return true
}
if matrix[i][j] < target {
j++
} else {
i--
}
}
return false
}
Python
class Solution:
def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
m, n = len(matrix), len(matrix[0])
i, j = m - 1, 0
while i >= 0 and j < n:
if matrix[i][j] == target:
return True
elif matrix[i][j] < target:
j += 1
else:
i -= 1
return False
时间复杂度:
O
(
m
+
n
)
O(m+n)
O(m+n)
空间复杂度:
O
(
1
)
O(1)
O(1)。