二维数组中的查找

Wu Jun 2020-01-14 21:30:12
01 数据结构与算法 > 算法题 > 数组

题目出处

二维数组中的查找

1 题目描述

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

Consider the following 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]
]

Given target = 5, return true.
Given target = 20, return false.

2 解题思路

要求时间复杂度 O(M + N),空间复杂度 O(1)。其中 M 为行数,N 为 列数。

该二维数组中的一个数,小于它的数一定在其左边,大于它的数一定在其下边。因此,从右上角开始查找,就可以根据 target 和当前元素的大小关系来缩小查找区间,当前元素的查找区间为左下角的所有元素。

public class Solution {
    public boolean Find(int target, int [][] array) {
        if(null == array || null == array[0] || 1 > array[0].length){
            return false;
        }
        int rowLength = array.length;
        int cowLength = array[0].length;
        for(int i = 0, j = cowLength - 1; i < rowLength && j >= 0;){
            int currentValue = array[i][j];
            if(currentValue == target){
                return true;
            }else if(currentValue > target){
                j--;
            }else{
                i++;
            }
        }
        return false;
    }
}

3 举一反三

遇到复杂问题时,通过具体的例子找出其中的规律。