当前位置:  首页>> 技术小册>> 数据结构与算法(下)

题目描述

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

解法

从二维数组的右上方开始查找:

  • 若元素值等于 target,返回 true
  • 若元素值大于 target,砍掉这一列,即 --j
  • 若元素值小于 target,砍掉这一行,即 ++i

也可以从二维数组的左下方开始查找,以下代码使用左下方作为查找的起点。

注意,不能选择左上方或者右下方的数字,因为这样无法缩小查找的范围。

  1. /**
  2. * @author bingo
  3. * @since 2018/10/27
  4. */
  5. public class Solution {
  6. /**
  7. * 二维数组中的查找
  8. * @param target 目标值
  9. * @param array 二维数组
  10. * @return boolean
  11. */
  12. public boolean find(int target, int[][] array) {
  13. if (array == null) {
  14. return false;
  15. }
  16. int rows = array.length;
  17. int columns = array[0].length;
  18. int i = rows - 1;
  19. int j = 0;
  20. while (i >= 0 && j < columns) {
  21. if (array[i][j] == target) {
  22. return true;
  23. }
  24. if (array[i][j] < target) {
  25. ++j;
  26. } else {
  27. --i;
  28. }
  29. }
  30. return false;
  31. }
  32. }

测试用例

  1. 二维数组中包含查找的数字(查找的数字是数组中的最大值和最小值;查找的数字介于数组中的最大值和最小值之间);
  2. 二维数组中没有查找的数字(查找的数字大于/小于数组中的最大值;查找的数字在数组的最大值和最小值之间但数组中没有这个数字);
  3. 特殊输入测试(输入空指针)。

该分类下的相关小册推荐: