题目:
编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。
示例 1:
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 3
输出:true
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-a-2d-matrix
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
方法1: 暴力
方法2: 看到排序 + 索引,直接二分。
方法3: 偷师三叶的,抽象BST解法。
设右上角为头节点,构建BST树。
//C语言暴力
bool searchMatrix(int** matrix, int matrixSize, int* matrixColSize, int target){
for(int i=0;i<matrixSize;i++){
for(int j=0;j<*matrixColSize;j++){
if(matrix[i][j]==target)
return true;
}
}
return false;
}
//典型的二分法
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
//special
if (matrix == null) {
return false;
}
int m = matrix.length;
int n = matrix[0].length;
int l = 0;
int r = m * n - 1;
while (l <= r) {
int mid = l + (r - l )/ 2;
if (matrix[mid / n][mid % n] == target) {
return true;
} else if (matrix[mid / n][mid % n] > target) {
r = mid - 1;
} else {
l = mid + 1;
}
}
return false;
}
}
//抽象BST
class Solution {
int m, n;
public boolean searchMatrix(int[][] mat, int t) {
m = mat.length; n = mat[0].length;
int x = 0, y = n - 1;
while (check(x, y) && mat[x][y] != t) {
if (mat[x][y] > t) {
y--;
} else {
x++;
}
}
return check(x, y) && mat[x][y] == t;
}
boolean check(int x, int y) {
return x >= 0 && x < m && y >= 0 && y < n;
}
}