Leetcode中级算法-二分查找01

题意:原题链接

编写一个高效的算法来搜索 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;
    }
};

猜你喜欢

转载自blog.csdn.net/liushengxi_root/article/details/81261452