剑指offer JZ1 - 二维数组中的查找 - JavaScript

1、题目描述

在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。

请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

2、解题思路

解法一：遍历数组中的所有元素，找到是否存在。时间复杂度是 O(N^2)，空间复杂度是 O(1)。

function Find(target, array)
{
    let flag = false;
    array.map(item=>{
        item.map(secondItem=>{
            if(secondItem === target){
                flag = true;
            }
        })
    })
    return flag;
}

解法二：按照数组的特点：每行都按照从左到右递增的顺序排序，每列都按照从上到下递增的顺序排序。考虑以下数组：

1 2 3
4 5 6
7 8 9

在其中寻找 5 是否存在。过程如下：

• 从右上角开始遍历
• 当前元素小于目标元素(3 < 5)，根据数组特点，当前行中最大元素也小于目标元素，因此进入下一行
• 当前元素大于目标元素(6 > 5)，根据数组特点，行数不变，尝试向前一列查找
• 找到 5

代码如下：

function Find(target, array) {
    const rowNum = array.length;    //一维数组的个数
    if (!rowNum) {
        return false;
    }
    const colNum = array[0].length;    //每个一维数组的长度
    if (!colNum) {
        return false;
    }
 
    let row = 0,
        col = colNum - 1;
    while (row < rowNum && col >= 0) {    //从每一一维数组的末尾进行查找
        if (array[row][col] === target) {
            return true;
        } else if (array[row][col] > target) {
            --col;
        } else {
            ++row;
        }
    }
    return false;
}

时间复杂度是 O(M+N)，空间复杂度是 O(1)。其中 M 和 N 分别代表行数和列数。