搜索二维矩阵

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

每行中的整数从左到右按升序排列。

每行的第一个整数大于前一行的最后一个整数。

示例 :

输入:

matrix = [

  [1,  3,  5,  7],

  [10, 11, 16, 20],

  [23, 30, 34, 50]

]

target = 3

输出: true

代码

class Solution {

  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;

  }

}

作者:LeetCode

链接:https://leetcode-cn.com/problems/search-a-2d-matrix/solution/sou-suo-er-wei-ju-zhen-by-leetcode/

来源:力扣(LeetCode)

著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

友情链接更多精彩内容