剑指offer系列(2)——二维数组中的查找

来源:互联网 发布:矩阵向量化公式 编辑:程序博客网 时间:2024/06/08 18:55

题目如下:
在一个二维数组中,每一行都按照从左到右递增的顺序排序。每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
示例如下:
1 2 8 9
2 4 9 12
4 7 10 13
6 8 11 15
当查找数字7时,返回true,当查找数字5,由于数组不包含此数字,则返回false;
普通的思路为:当数组中选取的数字与要求数字相等时,停止查找,返回true,当选取的数字小于要求数字时,扫描选取数字的下方或右方,当选取的数字大于要求数字时,扫描选取数字的左方或上方。这样的话,每次选择下一个时就有两个可能的方向,而这两个方向又会有重叠的部分,所以处理起来很麻烦。

我们可以从二维数组的右上角或者左下角来开始比较(这两个点的共同点为他们的一个方向比他小,另一个方向比他大,这样的话无论判断结果是什么,都会舍去一个方向。)这里以右上角为例:
具体思路为:当数组中选取的数字与要求数字相等时,停止查找,返回true,当选取的数字小于要求数字时,扫描选取数字的下方,当选取的数字大于要求数字时,扫描选取数字的左方。
代码实现如下:

public class Two_DimensionalArray {    public static void main(String[] args) {        int[][] a = {{1,2,8,9},{2,4,9,12},{4,7,10,13},{6,8,11,15}};//new int[4][4]{1,2,8,9,2,4,9,12,4,7,10,13,6,8,11,15};        System.out.println(findTwoArray(a,5));    }    public static boolean findTwoArray(int[][] obj , int n){        if (obj == null ) {            return false;        }        int i = 0;  //行变量        int j = obj[0].length - 1;;  //列变量        while (i < obj.length && j < obj[0].length) {            //obj.length:第一维数组长度            //obj[0].length:第二维数组长度            if (obj[i][j] > n) {                j--;            }else if(obj[i][j] < n){                i++;            }else{                return true;            }            if (i > obj.length || j < 0) {                //已知矩阵为方阵                return false;            }        }        return false;    }}

运行结果:
查询7时返回true;
查询5时返回false;

注意:本代码只适用于矩阵为方阵,否则还需判断是否可以往下或往左扫描!

原创粉丝点击