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 =
» Solve this problem3
, return true
.[Thoughts]
做两次二分就好了,首先二分第一列,找出target所在的行,然后二分该行。
[Code]
1: bool searchMatrix(vector<vector<int> > &matrix, int target) { 2: int row = matrix.size(); 3: if(row ==0) return false; 4: int col = matrix[0].size(); 5: if(col ==0) return false; 6: if(target< matrix[0][0]) return false; 7: int start = 0, end = row-1; 8: while(start<
=
end) 9: { 10: int mid = (start+end)/2; 11: if(matrix[mid][0] == target) 12: return true; 13: else if(matrix[mid][0] < target) 14: start = mid+1; 15: else 16: end = mid-1; 17: } 18: int targetRow = end; 19: start =0; 20: end = col-1; 21: while(start <
=
end) 22: { 23: int mid = (start+end)/2; 24: if(matrix[targetRow][mid] == target) 25: return true; 26: else if(matrix[targetRow][mid] < target) 27: start = mid+1; 28: else 29: end = mid-1; 30: } 31: return false; 32: }
注意,
二分的条件应该是(start<=end), 而不是(start<end)。
No comments:
Post a Comment