二维数组中的查找

题目描述

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

1	3	5	7	9	15
4	8	10	12	15	17
7	9	15	19	20	21
9	15	20	25	29	35

解题思路

从右上角开始，目标值若比右上角值小，向左走，删除右上角所在那一列；目标值若比右上角值大，向下走，删除右上角所在那一行。

如下图所示，若目标值为8，则比较8 < 15 ，15（右上角）所在列可以忽略；若目标值为21，则比较21 > 15，15（右上角）所在行可以忽略。

解题代码

核心方法

    /**
     * 从右上角开始找起
     *      如果当前值比右上角值小，则右上角那列可删除
     *      如果当前值比右上角值大，则右上角那行可删除
     */
    public static boolean find(int target, int [][] array) {
        // 右上角初始坐标
        int rows = 0;
        int cols = array[rows].length - 1;
        // 循环遍历，当行或列超出则没有检索到相同元素
        while (rows <= (array.length - 1) && cols >= 0) {
            if (target == array[rows][cols]) {
                return true;
            } else if (target > array[rows][cols]) {
                rows++;
            } else {
                cols--;
            }
        }
        return false;
    }

完整案例 

import java.util.*;
public class TwoArrayFind {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int m = sc.nextInt();
        int n = sc.nextInt();
        int[][] arr = new int[m][n];

        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                arr[i][j] = sc.nextInt();
            }
        }
        int target = sc.nextInt();
        System.out.println(find(target,arr));
    }
    /**
     * 从右上角开始找起
     *      如果当前值比右上角值小，则右上角那列可删除
     *      如果当前值比右上角值大，则右上角那行可删除
     */
    public static boolean find(int target, int [][] array) {
        // 右上角初始坐标
        int rows = 0;
        int cols = array[rows].length - 1;
        // 循环遍历，当行或列超出则没有检索到相同元素
        while (rows <= (array.length - 1) && cols >= 0) {
            if (target == array[rows][cols]) {
                return true;
            } else if (target > array[rows][cols]) {
                rows++;
            } else {
                cols--;
            }
        }
        return false;
    }
}