剑指offer(4)——二维数组中的查找
一、题目描述
在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
示例: 现在有矩阵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; }