JZ1-二维数组中的查找

题目描述

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

样例输入输出

[
  [1,2,8,9],
  [2,4,9,12],
  [4,7,10,13],
  [6,8,11,15]
]
给定 target = 7，返回 true。
给定 target = 3，返回 false。

暴力法

直接两次for循环遍历，将所有元素全都遍历一次，找到就返回true，没有找到就返回false

public class Solution {
    public boolean Find(int target, int [][] array) {
            for (int i = 0; i < array.length; i++) {
            for (int j = 0; j < array[i].length; j++) {
                if(target == array[i][j]){
                    return true;
                }
            }
        }
        return false;
    }
}

复杂度分析

时间复杂度：O(n^2)
空间复杂度：O(1)

优秀解答

从右上角查

将右上角元素定义为m。
应该明确右上角元素m，是该行最大元素，该列最小元素，那么和target比较会出现以下三种情况：

• m>target，该列最小元素大于target，说明该列其他元素也一定比target大，所以将列号减1进行下次判断；
• m<target，该行最大元素小于target，说明该行的其他元素也一定比target小，所以将行号加1进行下次判断；
• m==target，找到target，直接返回true

代码实现

public class Solution {
    public boolean Find(int target, int [][] array) {
        int rowSum = array.length;//array的总行数
        if(rowSum == 0){
            return false;
        }
        int columnSum = array[0].length;//array的总列数
        int column = columnSum - 1;//右上角元素index
        int row = 0;
        while(column >=0 && row < rowSum){//防止数组越界
            if(array[row][column] > target){//该列最小元素大于target
                column--;//列号减1
            }else if(array[row][column] < target){//该行最大元素小于target
                row++;//行号加1
            }else{
                return true;//找到target，返回true
            }
        }
        return false;//没有找到target，返回false
    }
}

复杂度分析

时间复杂度：O(行高 + 列宽)
空间复杂度：O(1)

从左下角查

将左下角元素定义为m。
应该明确左下角元素m，是该行最小元素，该列最大元素，那么和target比较会出现以下三种情况：

• m>target，该行最小元素大于target，说明该行其他元素也一定比target大，所以将行号减1进行下次判断；
• m<target，该列最大元素小于target，说明该列的其他元素也一定比target小，所以将列号加1进行下次判断；
• m==target，找到target，直接返回true

代码实现

public class Solution {
    public boolean Find(int target, int [][] array) {
        int rowSum = array.length;//array的总行数
        if(rowSum == 0){
            return false;
        }
        int columnSum = array[0].length;//array的总列数
        int column = 0;//右上角元素index
        int row = rowSum - 1;
        while(row >=0 && column < columnSum){//防止数组越界
            if(array[row][column] > target){//该行最小元素大于target
                row--;//行号减1
            }else if(array[row][column] < target){//该列最大元素小于target
                column++;//列号加1
            }else{
                return true;//找到target，返回true
            }
        }
        return false;//没有找到target，返回false
    }
}

复杂度分析

时间复杂度：O(行高 + 列宽)
空间复杂度：O(1)

思路分析

该题主要抓住每行每列以升序排序，可以得到两个位置比较特殊的元素，右上角元素（该行中最大，该列中最小），左下角元素（该行中最小，该列中最大）。根据这两个元素和target比较出现的不同情况进行行号和列号的调整，同时要保证不能数组越界。