74. Search a 2D Matrix

題目鏈接:74

思路

由題意可知:每一行的第一個(gè)元素都比上一行的第一列元素大,也即第一列是遞增的。因此可以使用兩次binary search來找到答案。

首先對第一列進(jìn)行binary search,找出第一列中不大于 target 的最大的元素;然后對該元素所在的行進(jìn)行 binary search,找出是否存在 target。

// 1. do binary search on first column, find the biggest element that is not bigger than target;
// 2. do binary search on the row of the element, find whether target exist
func searchMatrix(matrix [][]int, target int) bool {
    // 1.
    left, right := 0, len(matrix) - 1
    for left < right - 1 {
        mid := left + (right - left) / 2
        if matrix[mid][0] <= target {
            left = mid
        } else {
            right = mid - 1
        }
    }
    var eleR int
    if matrix[left][0] > target {
        return false
    } else if matrix[right][0] <= target {
        eleR = right
    } else {
        eleR = left
    }

    // 2.
    left, right = 0, len(matrix[eleR]) - 1
    for left <= right {
        mid := left + (right - left) / 2
        if matrix[eleR][mid] == target {
            return true
        } else if matrix[eleR][mid] < target {
            left = mid + 1
        } else {
            right = mid - 1
        }
    }
    return false
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容