查找二维矩阵

查找二维矩阵

题目参考 力扣

在一个二维矩阵中查找一个数，第一个想到的是一层一层遍历，如果数据够大，这样查找是否包含一个数据就很废时间，如果没有数据就会时间复杂度到达 O(M*N) M*N为数据个数

 

当然这道题主要是针对已经排序好的二维数组来计算的。也就是每一行从左到右升序，每一行的第一个数比上一行最后一个数大。

在这样的二维矩阵中查找数据，我们可以将其看作是一个一维的数组，可以使用二分查找法来进行查找数据

图解

 

下面我们用算法实现

 public boolean searchMatrix(int[][] matrix, int target) {
    int m = matrix.length;
    if (m == 0) return false;
    int n = matrix[0].length;

    // 二分查找
    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;
  }