题目描述

描述转载自力扣《74. 搜索二维矩阵》
编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:

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

示例1

图片说明

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

示例2

图片说明

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

解题思路

  1. 比较 target 和每一行最后一个元素,并截断为一维数组
    图片说明
    图片说明
    图片说明
  2. 使用二分查找

Java 代码实现

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        int rowLen = matrix.length;
        int colLen = matrix[0].length;
        int targetRow = 0;
        for (int i = 0; i < rowLen; ++i) {
            if (matrix[i][colLen - 1] > target) {
                targetRow = i;
                break;
            }
            else if(matrix[i][colLen - 1] == target) return true;
        }
        int[] arr = matrix[targetRow];
        int left = 0, right = colLen - 1, mid = 0;
        while (left <= right) {
            mid = (left + right) >> 1;
            if (arr[mid] == target) return true;
            else if (arr[mid] > target) right = mid - 1;
            else left = mid + 1;
        }
        return false;
    }
}