mirror of
https://github.com/ShusenTang/LeetCode.git
synced 2024-09-02 14:20:01 +00:00
25 lines
877 B
Markdown
25 lines
877 B
Markdown
![]() |
# [74. Search a 2D Matrix](https://leetcode.com/problems/search-a-2d-matrix/)
|
|||
|
# 思路
|
|||
|
就是一个二分查找。将矩阵展开成一个很长的有序数组,下标low从0开始,下标high从`m*n - 1`开始,然后二分查找,中间下标mid对应的元素为`matrix[mid / n][mid % n]`。
|
|||
|
时间复杂度O(log(mn)),空间复杂度O(1)
|
|||
|
|
|||
|
# C++
|
|||
|
``` C++
|
|||
|
class Solution {
|
|||
|
public:
|
|||
|
bool searchMatrix(vector<vector<int>>& matrix, int target) {
|
|||
|
if(matrix.empty()) return false;
|
|||
|
int m = matrix.size(), n = matrix[0].size();
|
|||
|
int low = 0, high = m * n - 1;
|
|||
|
while(low <= high){
|
|||
|
int mid = (high - low) / 2 + low;
|
|||
|
int tmp = matrix[mid / n][mid % n];
|
|||
|
if(tmp == target) return true;
|
|||
|
if(tmp < target) low = mid + 1;
|
|||
|
else high = mid - 1;
|
|||
|
}
|
|||
|
return false;
|
|||
|
}
|
|||
|
};
|
|||
|
```
|