LeetCode74. 搜索二维矩阵
编写一个高效的算法来判断 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
解题思路:
这个题目刚开始一直没有弄清楚题目的意思。题目的意思是在矩阵中找一个符合条件的target。但是这个矩阵有一个特点,就是上面提到的
1. 每行中的整数从左到右按升序排列。
1. 每行的第一个整数大于前一行的最后一个整数。
这个条件一看到的话首先想到的是:有序+查找,立马想到了二分查找算法。事实上题目里的官方解答就是二分查找。
但是这个题目还有一个解法,其实是比较通用的解法,也是普通人(比如楼主)能够想到的方法。就是利用给定的矩阵的数字特点,采用遍历的方法。下面分别给出这两种算法的代码。
解法1:
从最左下角开始遍历查找,比它大的元素在右边,比它小的元素在上边
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
if(matrix==null || matrix.length == 0 || matrix[0].length == 0) return false;
int row=matrix.length; //行
int colum=matrix[0].length; //列
int i=row-1;
int j=0; //从左下角开始找
while(i>=0 && j<colum ){
if(matrix[i][j]>target){
i--;
}
else if(matrix[i][j]<target){
j++;
}
else{
return true;
}
}
return false;
}
}
解法2:二分查找算法,这里面可以将二维矩阵退化为一维数组,注意这里的退化不是真正的创建一个一维数组,而是利用下标的关系来处理的。m x n 矩阵可以视为长度为 m x n的有序数组。由于该 虚数组的序号可以由下式方便地转化为原矩阵中的行和列 (我们当然不会真的创建一个新数组) ,该有序数组非常适合二分查找。
row = idx // n , col = idx % 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 left=0;
int right=m*n-1; //注意搜索范围时[left,right]
int index=0;
int indexElement=0;
while(left<=right){
index=left+(right-left)/2;
//注意下面的矩阵退化为数组的下标对应关系
indexElement=matrix[index/n][index%n];
if(target==indexElement){
return true;
}
else if(target<indexElement){
right=index-1;
}
else if(target>indexElement){
left=index+1;
}
}
return false;
}
}