《算法分析与设计》Week 10_write an efficient algorithm that searches for a v-优快云博客

本文链接：https://blog.youkuaiyun.com/Small_Hacker/article/details/72847866

本文介绍了一种高效的二维矩阵搜索算法，该算法适用于行和列都按升序排列且每行首元素大于前一行末元素的矩阵。通过将矩阵视为一维数组进行二分查找，实现O(Log(mn))的时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

74. Search a 2D Matrix

Description:

Write an efficient algorithm that searches for a value in an m x n matrix. This matrix has the following properties:

Integers in each row are sorted from left to right.
The first integer of each row is greater than the last integer of the previous row.

For example,

Consider the following matrix:

[
  [1,   3,  5,  7],
  [10, 11, 16, 20],
  [23, 30, 34, 50]
]

Given target = 3, return true.

Solution:

一、题意理解

在一个m x n 的二维矩阵中做一个查找算法。矩阵有下面的特点

(1) 每一行从左到右是升序的。

(2) 每一行第一个数比上一行最后一个数要大。

二、分析

1、直接查找的方式是对每一行执行一个二分查找，时间复杂度为 O(mLog(n))

2、仔细看一下题目中矩阵的特点，发现其完全可以当做一个一维矩阵，只需要将每一行首位连接即可，连接后的一维矩阵就是一个升序的数组，可以执行二分茶盏，时间复杂度为O(Log(mn))。这里所说的连接也不是真正的连接，只需要将一维数组的下标转化到二位矩阵的坐标即可。

3、代码如下：

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int row = matrix.size();
        if(row == 0)
            return false;
        int column = matrix[0].size();
        if(column == 0)
            return false;
        
        int begin = 0, end = row * column - 1;
        while(begin <= end)
        {
            int index = (begin + end) / 2;
            int num = matrix[index / column][ index % column];
            if(target == num)
                return true;
            else if(target < num)
                end = index - 1;
            else
                begin = index + 1;
        }
        return false;
    }
};