剑指 Offer 04. 二维数组中的查找

题目描述

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

示例1

• 输入：7,[[1,2,8,9],[2,4,9,12],[4,7,10,13],[6,8,11,15]]
• 返回值：true

难度：中等

---

题解

方法1: 暴力算法

思路： 直接遍历一遍数组，即可判断目标target是否存在。
复杂度分析：

• 时间复杂度：O(n^2)，因为最坏情况下，数组中的元素都需要遍历一次。
• 空间复杂度：O(1)

C#

class Solution
{
    public bool Find(int target, int[][] array)
    {
        for(int i=0;i<array.Length;i++) //arry.Length是二维数组的行数
        {
            for(int j=0;j<array[i].Length;j++) //array[i].Length是二维数组的列数
            {
                if(array[i][j]==target)
                    return true;
            }
        }
        return false;
    }
}

方法2:二分查找

        把二分值定在右上角或者左下角，就可以进行二分。
        由于给定的二维数组具备每行从左到右递增以及每列从上到下递增的特点，当访问到一个元素时，可以排除数组中的部分元素。
        从二维数组的右上角开始查找。如果当前元素等于目标值，则返回 true。如果当前元素大于目标值，则移到左边一列。如果当前元素小于目标值，则移到下边一行。

复杂度分析:

• 时间复杂度：O(n+m)。
  访问到的下标的行最多增加 n 次，列最多减少 m 次，因此循环体最多执行 n + m 次。
• 空间复杂度：O(1)。

C#:

class Solution
{
    public bool Find(int target, int[][] array)
    {
        bool found = false;
        
        int row = array.Length;// 行数
        int col = array[0].Length; //列数
        
        if(row ==0 || col ==0) // 二维数组为空，则直接false
        {
            return found;
        }
        
        int row_i = row-1;
        int col_j = 0;
        
        while(array!=null && row_i>=0 && col_j< col)
        {
            if(array[row_i][col_j] == target)
            {
                found =true;
                return found;
            }
            else if(array[row_i][col_j] >target) // 比target大，向左移
            {
                row_i--;
            }
            else // 比target小，向下移
            {
                col_j++;
            }
        }
        return found;
    }
}