Leetcode 面試題04. 二維數組中的查找【二分搜索】

問題描述

在一個 n * m 的二維數組中，每一行都按照從左到右遞增的順序排序，每一列都按照從上到下遞增的順序排序。請完成一個函數，輸入這樣的一個二維數組和一個整數，判斷數組中是否含有該整數。

示例:

現有矩陣 matrix 如下：

[
[1, 4, 7, 11, 15],
[2, 5, 8, 12, 19],
[3, 6, 9, 16, 22],
[10, 13, 14, 17, 24],
[18, 21, 23, 26, 30]
]
給定 target = 5，返回 true。

給定 target = 20，返回 false。

限制：

0 <= n <= 1000

0 <= m <= 1000

解題報告

從左上角或者右下角的元素開始搜索。
爲什麼可以從這兩個元素開始呢？
以右下角的元素爲例，它向上走，元素大小遞減，它向右走，元素大小遞增，兩個方向具有相反的單調性。左上角同理。

實現代碼

class Solution{
    public:
        bool findNumberIn2DArray(vector<vector<int>>&matrix, int target){
            if(matrix.size()==0) return false;
            int m=matrix.size(), n=matrix[0].size();
            int row=m-1,col=0;
            while(row>=0&&col<n){
                int num=matrix[row][col];
                if(num==target) return true;
                else if(num>target) row--;
                else col++;
            }
            return false;
        }
};

參考資料

[1] Leetcode 面試題04. 二維數組中的查找