首页 > 代码库 > Leetcode:Search a 2D Matrix
Leetcode:Search a 2D Matrix
Write an efficient algorithm that searches for a value in an m x n matrix. This matrix has the following properties:
- Integers in each row are sorted from left to right.
- The first integer of each row is greater than the last integer of the previous row.
For example,
Consider the following matrix:
[ [1, 3, 5, 7], [10, 11, 16, 20], [23, 30, 34, 50]]
Given target = 3
, return true
.
分析:将二维数组看为一个一维sorted数组,然后用bianry search。时间复杂度为O(log(m*n)),代码如下:
1 class Solution { 2 public: 3 bool searchMatrix(vector<vector<int> > &matrix, int target) { 4 int m = matrix.size(); 5 if(m == 0) return false; 6 int n = matrix[0].size(); 7 if(n == 0) return false; 8 9 int l = 0, r = m * n - 1;10 while(l <= r){11 int mid = (l + r)/2;12 int row = mid/n, col = mid%n;13 if(matrix[row][col] == target) return true;14 if(matrix[row][col] < target) l = mid + 1;15 else r = mid - 1;16 }17 18 return false;19 }20 };
Leetcode:Search a 2D Matrix
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。