LeetCode: 搜索二维矩阵

编写一个高效的算法来判断 m x n 矩阵中，是否存在一个目标值。该矩阵具有如下特性：

• 每行中的整数从左到右按升序排列。
• 每行的第一个整数大于前一行的最后一个整数。

示例 1:

输入:
matrix = [
  [1,   3,  5,  7],
  [10, 11, 16, 20],
  [23, 30, 34, 50]
]
target = 3
输出: true

示例 2:

输入:
matrix = [
  [1,   3,  5,  7],
  [10, 11, 16, 20],
  [23, 30, 34, 50]
]
target = 13
输出: false

 

// 两次二分查找
//执行用时 : 16 ms, 在Search a 2D Matrix的C++提交中击败了94.13% 的用户
//内存消耗 : 9.8 MB, 在Search a 2D Matrix的C++提交中击败了80.94% 的用户
class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int m = matrix.size();
        if(!m)
            return false;
        int n = matrix[0].size();
        if(!n)
            return false;
        int top = 0, down = m - 1;
        int left = 0, right = n - 1;
        int temp = 0;
        // 先查找属于哪一行 注意 <= 存在只有一个元素的情况 这个时候top==down
        while(top <= down){
            std::cout << top << down << endl;
            temp = (top + down) / 2;
            if(matrix[temp][0] == target)
                return true;
            else if(matrix[temp][0] < target)
                top = temp;
            else
                down = temp;
            //top==down 说明只有一个元素 这时退出即可
            //top==down-1 说明找到边界了 直接退出
            if(top==down || top == down-1)
                break;
        }
        // 对于 要找的目标大于矩阵所有值时 更新 所要查找的行
        if(top == down-1 && matrix[down][0] <= target)
            top = down;
        // 对于要找的目标小于矩阵所有值时 直接返回False
        if(matrix[top][0] > target)
            return false;
        //在一行查找目标值 就是简单的二分查找
        while(left < right){
            temp = (left + right) / 2;
            if(matrix[top][temp] == target)
                return true;
            else if(matrix[top][temp] < target)
                left = temp + 1;
            else
                right = temp -1;
        }
        // 最后要判断注意剩下的情况 就是left==right了
        if(matrix[top][left] == target)
            return true;
        return false;          
    }
};

// 从右上角开始查找 小于往左找 大于向下找  O(max(m,n))
// 执行用时 : 64 ms, 在Search a 2D Matrix的C++提交中击败了25.13% 的用户
// 内存消耗 : 9.9 MB, 在Search a 2D Matrix的C++提交中击败了73.34% 的用户
class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int m = matrix.size();
        if(!m)
            return false;
        int n = matrix[0].size();
        if(!n)
            return false;
        int i = 0, j = n -1;
        while(i < m && j >=0){
            std::cout << i << j <<endl;
            if(matrix[i][j] == target)
                return true;
            else if(matrix[i][j] < target)
                i++;
            else
                j--;
        }
        return false;
    }
};