題目
編寫一個高效的算法來搜索 m x n 矩陣 matrix 中的一個目標值 target。該矩陣具有以下特性:
每行的元素從左到右升序排列。
每列的元素從上到下升序排列。
示例:
現有矩陣 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。
思路
[
小< ——大
^
|
大
]
上圖就是題目給出的數字特點,利用這個特點,我們要找到一個方向可以向階梯一樣,一階一階的排除,此時就只有兩個方向,左下
和右上
方向。
說明:從右上向左下方向走,我們用一個[i,j]表示當前位置,從右上開始,當前值大於目標時,我們就向左走,當前值小於目標,我們就向下走,因爲在我們的左邊是比我們小的,在我們的下邊是比我們大的,我們可以走鋸齒狀,一列一行的排除。看下代碼就明白了呢。當然,從左下到右上的過程也是可以的。
解法
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
int row = matrix.length;
if(row == 0) return false; //特判
int col = matrix[0].length;
int i = 0,j = col-1;
while(i < row && j >= 0) {
if(matrix[i][j]<target) {
i++;
} else if(matrix[i][j] > target) {
j--;
} else{
return true;
}
}
return false;
}
}