leetcode-剑指 Offer 04. 二维数组中的查找
在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
示例:
现有矩阵 matrix 如下:
[
[1, 4, 7, 11, 15],
[2, 5, 8, 12, 19],
[3, 6, 9, 16, 22],
[10, 13, 14, 17, 24],
[18, 21, 23, 26, 30]
]
给定 target = 5,返回 true。
给定 target = 20,返回 false。
限制:
0 <= n <= 1000
0 <= m <= 1000
解题思路:
方法一: 暴力查找
将所有数字放到一维数组中,排序,然后在有序数组中二分查找target。
时间复杂度:O(nm)
空间复杂度:O(nm)
class Solution {
public:
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
vector<int> arry;
if(matrix.size()==0) return false;
int n=matrix.size(),m=matrix[0].size();
for(int i =0;i<n;i++)
for(int j=0;j<m;j++)
arry.push_back(matrix[i][j]);
sort(arry.begin(),arry.end());
int i=0,j=n*m-1;
while(i<=j)
{
int mid=i+(j-i)/2;
if(arry[mid]>target)
j=mid-1;
else if(arry[mid]<target)
i=mid+1;
else
return true;
}
return false;
}
};
方法二: 题目给出的数组是行有序、列有序的,所以方法一显然不是最优的
以nums[0] [n-1]作为起始点,
如果taget>nums[0] [n-1] i++
如果taget<nums[0] [n-1] j–
如果taget==nums[0] [n-1] 存在目标值
时间复杂度:O(n+m)
空间复杂度:O(1) 为i,j需要的空间
class Solution {
public:
bool findNumberIn2DArray(vector<vector<int>>& nums, int target) {
if(nums.size()==0) return false;
int m=nums.size(),n=nums[0].size();
int i=0,j=n-1;
while(i<m&&j>=0)
{
if(nums[i][j]>target) j--;
else if(nums[i][j]<target) i++;
else return true;
}
return false;
}
};