搜索二维矩阵 II

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-a-2d-matrix-ii

题目描述:

编写一个高效的算法来搜索 m x n 矩阵 matrix 中的一个目标值 target 。该矩阵具有以下特性：
每行的元素从左到右升序排列。
每列的元素从上到下升序排列。

示例 1：

输入：matrix = [[1,4,7,11,15],[2,5,8,12,19],[3,6,9,16,22],[10,13,14,17,24],[18,21,23,26,30]], target = 5
输出：true

示例 2：

输入：matrix = [[1,4,7,11,15],[2,5,8,12,19],[3,6,9,16,22],[10,13,14,17,24],[18,21,23,26,30]], target = 20
输出：false

思路:

二分法

• 将矩阵的每一行看作独立的,则变成了从一个有序的一维数组中查找某一个数.

代码实现:

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        int row = matrix.length;
        int col = matrix[0].length;
        // 对每一行进行二分查找
        for (int i = 0; i < row; i++) {
            int a = 0, b = col - 1;
            while (a < b) {
                int mid = (a + b + 1) / 2;
                if (matrix[i][mid] <= target) a = mid;
                else b = --mid;
            }
            if (matrix[i][b] == target) return true;
        }
        return false;
    }
}