题意:原题链接
编写一个高效的算法来搜索 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。
很显然,遍历该二位数组是最愚蠢的做法,如果这样做也是挺智障的。那有没有更好的想法呐?emmmmmmm,好吧,还是直说吧,就不废话了 。用我们刚刚讲过的二叉查找树的思路解决。
思路:从该二维数组右上角的数字开始,如果该数字和target相等,那么查找完毕;如果小于,往左走,如果大于往下走 。就像二叉查找树一样,就把二位数组看作一颗二叉查找树就行了!!!
源码:
class Solution
{
public:
bool searchMatrix(vector<vector<int>> &matrix, int target)
{
if (matrix.size() == 0 || matrix[0].size() == 0)
return false; //处理空的情况
auto left = matrix[0].size() - 1;
auto right = 0;
while (matrix[right][left] != target)
{
auto it = matrix[right][left];
if (it == target)
return true;
if (target < it)
{
if (left > 0 )
left--;
else
return false;
}
else // must target > it
{
if (right < matrix.size()-1 )
right++;
else
return false;
}
}
return true;
}
};