題目描述:給定一個二維數組,其每一行從左到右遞增排序,從上到下也是遞增排序。給定一個數,判斷這個數是否在該二維數組中。要求時間複雜度 O(M + N),空間複雜度 O(1)。其中 M 爲行數,N 爲 列數。
Consider the following 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]
]
Given target = 5, return true.
Given target = 20, return false.
解題思路:
二維數組中的一個數,小於它的數一定在其左邊,大於它的數一定在其下邊。因此。從左下角或者右上角開始查找,本文從右上角開始查找,從第一個最後一列開始於target進行比對,若target小於當前元素值,則向下(行數增加)一行查找,若target大於當前元素值,則向前(列數減少)一列查找。
bool Find(int target, int[][] matrix)
{
if (matrix == null || matrix.length == 0 || matrix[0].length == 0)
return false;
int rows = matrix.length, cols = matrix[0].length;
int r = 0, c = cols - 1; // 從右上角開始
while (r <= rows - 1 && c >= 0) {
if (target == matrix[r][c])
return true;
else if (target > matrix[r][c])
r++;
else
c--;
}
return false;
}