剑指offer 03：二维数组中的查找

程序员小浩

发布于 2020-08-10 15:01:47

6160

发布于 2020-08-10 15:01:47

文章被收录于专栏：小浩算法小浩算法小浩算法

❝永远要这样写代码，好像最终维护你代码的人是个狂暴的、知道你住在哪里的精神病患者—— 小浩算法 ❞

二维数组中的查找

题目描述

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

解法

从二维数组的右上方开始查找：

若元素值等于 target，返回 true；
若元素值大于 target，砍掉这一列，即 --j；
若元素值小于 target，砍掉这一行，即 ++i。

也可以从二维数组的左下方开始查找，以下代码使用左下方作为查找的起点。

注意，不能选择左上方或者右下方的数字，因为这样无法缩小查找的范围。


public class Solution {
    /**
     * 二维数组中的查找
     * @param target 目标值
     * @param array 二维数组
     * @return boolean
     */
    public boolean find(int target, int[][] array) {
        if (array == null) {
            return false;
        }
        int rows = array.length;
        int columns = array[0].length;
        
        int i = rows - 1;
        int j = 0;
        while (i >= 0 && j < columns) {
            if (array[i][j] == target) {
                return true;
            }
            if (array[i][j] < target) {
                ++j;
            } else {
                --i;
            }
        }
        return false;
    }
}