且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

剑指offer系列之三:在二维数组中查找元素

更新时间:2022-06-12 08:50:45

题目描述:

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

由于题目条件的成立,所以使得这道题可以使用对角线的方法完成,可以从右上角的元素考虑,如果目标查找元素小于右上角的元素,那么不可能在右上角元素所在的列,如果目标大于剩余列的右上角的元素,那么不可能在该右上角元素所在的行。依照这个规律,就可以完成目标元素的查找(参考剑指offer书中的思路)。基于此,我写出如下的代码(已被牛客OJ平台AC):

package com.rhwayfun.offer;

public class MartrixFoundSolution {

    public boolean find(int [][] array,int target) {
        boolean found = false;
        int rows = array.length;
        int columns = array[0].length;
        int row = 0;
        int column = columns - 1;
        while(row < rows && column >=0){
            if(array[row][column] == target){
                found = true;
                break;
            }else if(array[row][column] > target){
                column--;
            }else{
                row++;
            }
        }
        return found;
    }

    public static void main(String[] args) {
        int[][] array = new int[4][4];
        array[0] = new int[]{1,2,8,9};
        array[1] = new int[]{2,4,9,12};
        array[2] = new int[]{4,7,10,13};
        array[3] = new int[]{6,8,11,15};

        boolean found = new MartrixFoundSolution().find(array, 7);
        System.out.println(found);
    }
}