leetcode 74. 搜索二維矩陣【二分】

題目鏈接:74. 搜索二維矩陣

編寫一個高效的算法來判斷 m x n 矩陣中,是否存在一個目標值。該矩陣具有如下特性:

每行中的整數從左到右按升序排列。
每行的第一個整數大於前一行的最後一個整數。

示例 1:

輸入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 3
輸出: true

示例 2:

輸入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 13
輸出: false

程序說明:

將二維數組映射爲一維數組,下標之間的關係:
row = index / n
col = index % n

代碼如下:

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        int m = matrix.length;
        if(m == 0)
            return false;
        int n = matrix[0].length;
        int l = 0, r = m * n - 1;

        while(l <= r) {
            int mid = (l + r) / 2;
            int tmp = matrix[mid / n][mid % n];
            if(target == tmp)
                return true;
            else if(target < tmp)
                r = mid - 1;
            else
                l = mid + 1;
        }
        return false;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章