題目鏈接:74. 搜索二維矩陣
編寫一個高效的算法來判斷 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
程序說明:
將二維數組映射爲一維數組,下標之間的關係:
row = index / n
col = index % n
代碼如下:
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
int m = matrix.length;
if(m == 0)
return false;
int n = matrix[0].length;
int l = 0, r = m * n - 1;
while(l <= r) {
int mid = (l + r) / 2;
int tmp = matrix[mid / n][mid % n];
if(target == tmp)
return true;
else if(target < tmp)
r = mid - 1;
else
l = mid + 1;
}
return false;
}
}