Leetcode 搜索二维矩阵

搜索二维矩阵

题目描述：

编写一个高效的算法来判断 m x n 矩阵中，是否存在一个目标值。该矩阵具有如下特性：

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

提示：

   m == matrix.length

   n == matrix[i].length

   1 <= m, n <= 100

   -10 ^4 <= matrix[i][j], target <= 10 ^4

题目链接

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        int m = matrix.length;
        int n = matrix[0].length;
        // 首先找到该目标值所在矩阵行
        int row = -1;
        for(int i = 0 ; i<m ; i++){
            if(matrix[i][0]<=target && target<=matrix[i][n-1]){
                row = i;
                break;
            }
        }
        if(row<0){
            return false;
        }
        if(m == n && m == 1) return true; // 特判：一个元素
        // 通过二分查找该行元素
        int left = 0;
        int right = n-1;
        while(left<=right){
            int mid = (left+right)/2;
            if(matrix[row][mid] < target ){
                left = mid+1;
            }else if(matrix[row][mid] > target){
                right = mid-1;
            }else{
                return true;
            }
        }
        return false;
    }
}

该题思路还是挺明确的，先遍历矩阵行，找到目标值所在行，如果未找到则直接返回false，如果能找到，就利用二分查找算法对该行进行二分查找。