lintcode 28 搜索二维矩阵

lintcode 28 搜索二位矩阵

写出一个高效的算法来搜索 m × n矩阵中的值。

这个矩阵具有以下特性:

每行中的整数从左到右是排序的。
每行的第一个数大于上一行的最后一个整数。
样例
样例 1:
输入: [[5]],2
输出: false

样例解释: 

没有包含,返回false。

样例 2:
输入:
[
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
],3
输出: true

样例解释: 
包含则返回true。

挑战
O(log(n) + log(m)) 时间复杂度

class Solution {
public:
    /**
     * @param matrix: matrix, a list of lists of integers
     * @param target: An integer
     * @return: a boolean, indicate whether matrix contains target
     */
    bool searchMatrix(vector<vector<int>> &matrix, int target) {
        // write your code here
        if(target<matrix[0][0]||target>matrix[matrix.size()-1][matrix[0].size()-1])
            return false;
        int low = 0;
        int high = matrix.size()-1;
        int mid,j;
        int left = 0;
        int right = matrix[0].size()-1;
        while(low<=high){
            mid = low + (high-low)/2;
            if(target==matrix[mid][0])
                return true;
            else if(target>matrix[mid][0])
                low = mid+1;
            else
                high = mid-1;
        }
        if(target<matrix[mid][0])
            j = mid-1;
        else
            j = mid;
        while(left<=right){
            mid = left + (right-left)/2;
            if(target==matrix[j][mid])
                return true;
            else if(target>matrix[j][mid])
                left = mid+1;
            else
                right = mid-1;
        }
        return false;
    }
};

遭遇问题:细节问题
if(targetmatrix[matrix.size()-1][matrix[0].size()-1])
return false;
一开始,把下标写错了,忘记用size()-1

你可能感兴趣的:(lintcode简单题,c++)