每日LeetCode一道题————搜索二维矩阵

每日一题

题目说明

240.搜索二维矩阵 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:
    int binarySearch(vector<int> nums, int target)
    {
        int start = 0;
        int end = nums.size() - 1;
        while (start <= end)
        {
            int mid = (start + end) / 2;
            if (target > nums[mid])
            {
                start = mid + 1;
            }
            else if (target < nums[mid])
            {
                end = mid - 1;
            }
            else
            {
                return mid;
            }
        }
        return -1;
    }
    bool searchMatrix(vector<vector<int>>& matrix, int target) 
    {
        if (matrix.size() == 0 || matrix[0].size() == 0)
        {
            return false;
        }
        for (int i = 0; i < matrix.size(); i++)
        {
            if (matrix[i][0] > target)
            {
                break;
            }
            if (matrix[i][matrix[i].size() - 1] < target)
            {
                continue;
            }
            int mid = binarySearch(matrix[i], target);
            if (mid != -1)
            {
                return true;
            }
        }
        return false;
    }
};

复杂度分析

• 时间复杂度：O(mlogn)。
• 空间复杂度：O(1)。