剑指Offer04:二维数组中的查找

剑指Offer04:二维数组中的查找

Scroll Down

剑指Offer04:二维数组中的查找

题目

题目链接

剑指Offer:二维数组中的查找

题目描述

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

分析

对于这道题目,直接进行数组的遍历,复杂度太高。

于是换一种思路,首先选取数组中右上角的数字。如果该数字等于要查找的数字,则查找过程结束;如果该数字大于要查找的数字,则剔除这个数字所在的列;如果该数字小于要查找的数字,则剔除这个数字所在的行。也就是说,如果要查找的数字不咋数组的右上角,则每次都在数组的查找范围中提出一行或一列,这样每一步都可以缩小查找的范围,直到找到要要查找的数字,或者查找范围为空。

20200217151115

代码

此代码是从右上找:

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

从左下找思路是一样的:

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