剑指offer——二维数组中的查找

剑指offer刷题总结——二维数组中的查找

 

题目描述

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

示例1

输入
7,[[1,2,8,9],[2,4,9,12],[4,7,10,13],[6,8,11,15]]


返回值
true

      这道题首先想到的肯定是暴力搜索，两层for循环，这样当然能够实现，而且这道题的空间限制和时间限制也够用。但是因为是算法题，我们需要考虑一下时间、空间复杂度。如果用两层for循环遍历查找，时间复杂度为O(M*N)

解1：

把二维数组的每一行看成是一维数组，然后进行二分查找。二分查找的时间复杂度是

public class Solution {
    public boolean Find(int target, int [][] array) {
        for (int i = 0; i < array.length; i++) {
            int low = 0;
            int high = array[i].length - 1;
            while(low <= high) {
                int mid = (low + high)/2;
                if(array[i][mid] == target)
                    return true;
                else if (array[i][mid] < target)
                    low = mid + 1;
                else
                    high = mid - 1;
            }
        }
        return false;
    }
}

解2：

利用二维数组从左至右、从上至下都是递增的特点，可以将target与二维数组的左下角或者右上角进行比较，我是与右上角进行比较。

如果target < a[row][col]，说明target一定在a[row][col]的左侧，所以col--；

如果target > a[row][col]，说明target一定在a[row][col]的下侧，所以row++

时间复杂度为O（N+M）

public class Solution {
    public boolean Find(int target, int [][] array) {
        int row = 0;
        int col = array[0].length - 1;
        
        while(row < array.length && col >= 0) {
            if(target == array[row][col])
                return true;
            else if (target < array[row][col]) {
                col--;
            } else {
                row++;
            }
        }
        return false;
    }
}