剑指 Offer 04. 二维数组中的查找

在一个 n * m 的二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个高效的函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法一：线性查找

根据给定的二维数组的特殊的性质，我们将其旋转45°进行观察，有点像二叉搜索树的结构。

选左下角的值为我们的基准值

• 如果当前遍历的元素比目标值大，那么行 - 1
• 如果当前遍历的元素比目标值小，那么列 + 1
• 如果当前遍历的元素与目标值相等，那么直接返回 true

#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
	bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
		//如果给定的数组它的行数为 0，即二维数组为空，直接返回 false 
		//为什么不是行和列都为 0，才返回 false 呢？因为列数需要用到 matrix[0]，如果二维数组为空，那么matrix[0] 的值就为空，获取 matrix[0].size() 会出现空指针的错误
		int linenums = matrix.size();
		if (linenums == 0) return false;
		int rownums = matrix[0].size();
		//从左下角开始寻找 target
		int i = linenums - 1;
		int j = 0;
		//当 i 和 j 所表示的行数，列数满足二维数组的范围时，进行比较
		while (i >= 0 && j < rownums) {
			//如果当前元素比目标值大，那么行 - 1
			if (matrix[i][j] > target) --i;
			//如果当前元素比目标值小，那么列 + 1
			else if (matrix[i][j] < target) ++j;
			//如果当前元素比目标值相等，那么直接返回 true
			else return true;
		}
		//执行到这里说明目标值不存在，直接返回 false 
		return false;
	}
};
int main()
{
	Solution test;
	vector<vector<int>> 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} };
	int target = 5;
	bool ans = test.findNumberIn2DArray(matrix, target);
	cout << ans << endl;
	return 0;
}

时间复杂度：O（n + m）。访问到的下标的行最多减少 n 次，列最多增加 m 次，因此循环体最多执行 n + m 次。

空间复杂度：O（1）。

方法二：二分查找

不建议这种方法来解决本题，因为没有用到本题涉及到的二维数组的从左到右、从上到下递增的特点，效率较低。

思路：在主方法中用 for 循环依次调用二分查找的函数，将二维数组按行分解成一个个单个的数组，对每一单个的数组进行二分查找。

class Solution {
public:
    //常规进行二分查找的函数
    bool search(vector<int>& nums, int target) {
        int left = 0;
        int right = nums.size() - 1;
        int mid = 0;
        while(left <= right) {
            mid = left + (right - left) / 2;
            if(nums[mid] == target) {
                return true;
            }
            else if(nums[mid] < target) {
                left = mid + 1;
            }
            else {
                right = mid - 1;
            }
        }
        return false;
    }
    //主方法中，一次将二维数组按行拆解为单个的一维数组，然后对一维数组进行二分查找
    bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
        if(matrix.size() == 0 || matrix[0].size() == 0){
            return false;
        }
        int row = matrix.size();
        for(int i = 0; i < row; ++i) {
            if(search(matrix[i], target)){
                return true;
            }
        }
        return false;
    }
};