搜索二维数组

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_39360985/article/details/84034196

编写一个高效的算法来判断 m x n 矩阵中，是否存在一个目标值。

该矩阵具有如下特性：

• 每行中的整数从左到右按升序排列。
• 每行的第一个整数大于前一行的最后一个整数。

示例 1:

输入:
matrix = [
  [1,   3,  5,  7],
  [10, 11, 16, 20],
  [23, 30, 34, 50]
]
target = 3
输出: true

示例 2:

输入:
matrix = [
  [1,   3,  5,  7],
  [10, 11, 16, 20],
  [23, 30, 34, 50]
]
target = 13
输出: false

题目分析

根据此矩阵的特性，每一行的第一个元素比前一行的最后一个元素大，并且每一行中元素从小到大排序，所以相当于一个从小到大排列的数字序列存在于二维数组中，要查找目标值的话，首先可以用目标值和第一行最后一个元素比较，如果相等，返回true；如果目标值小于此数，那么就在这一行寻找；如果目标值大于此数，则令行数增加，继续寻找。

代码实现

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

主函数

public static void main(String[] args)
{
   S7 s = new S7();
   boolean res1 = s.searchMatrix(new int[][]{{1,3,5,7},{10,11,16,20},{23,30,34,50}}, 3);
   boolean res2 = s.searchMatrix(new int[][]{{1,3,5,7},{10,11,16,20},{23,30,34,50}}, 13);
   System.out.println(res1);
   System.out.println(res2);
}

运行结果

true
false