剑指offer(4)——二维数组中的查找

227 阅读1分钟

一、题目描述

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

示例: 现在有矩阵matrix如下:

[
	[1,2,8 ,9 ],
	[2,4,9 ,12],
	[4,7,10,13],
	[6,8,11,15]
]

查找数字7,返回true;查找数字5,返回false。

二、解题思路

因为二维数组中从左到右,从上到下都是递增的,从右上角开始遍历,如果target < 右上角的元素,那么右上角所在的列可以排除掉;如果target > 右上角的元素,那么可以排除所在的行,直到找到目标元素。

代码实现

public boolean findNumberIn2DArray(int[][] matrix, int target) {
    if(matrix.length == 0)
        return false;
    int j = matrix[0].length - 1;
    int i = 0;
    while(j >= 0 && i < matrix.length){
        if(target < matrix[i][j]){
            j--;
        }else if(target > matrix[i][j]){
            i++;
        }else{
            return true;
        }
    }
    return false;
}