LeetCode: 74. 搜索二维矩阵(中等)

题目参考

编写一个高效的算法来判断 m x n 矩阵中，是否存在一个目标值。该矩阵具有如下特性：
每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。

题解

二分法
将m x n二维数组转化为一维m*n的数组。

row = idx // n ， col = idx % n。

class Solution {
  public:
  bool searchMatrix(vector<vector<int>>& matrix, int target) {
    int m = matrix.size();
    if (m == 0) return false;
    int n = matrix[0].size();

    // 二分查找
    int left = 0, right = m * n - 1;
    int pivotIdx, pivotElement;
    while (left <= right) {
      pivotIdx = (left + right) / 2;
      pivotElement = matrix[pivotIdx / n][pivotIdx % n];
      if (target == pivotElement) return true;
      else {
        if (target < pivotElement) right = pivotIdx - 1;
        else left = pivotIdx + 1;
      }
    }
    return false;
  }
};

复杂度分析

• 时间复杂度 : 时间复杂度为O(log(m n))。
• 空间复杂度 : O(1)。

参考

搜索二维矩阵
字节题库 - #74 - 中等 - 搜索二维矩阵 - 2刷
一文带你搞定多个二分查找变种题目