LeetCode Search a 2D Matrix（二分查找）

 

 

题意：

　　有一个矩阵，每行都有序，每行接在上一行尾后仍然有序。在此矩阵中查找是否存在某个数target。

 

思路：

　　这相当于用一个指针连续扫二维数组一样，一直p++就能到最后一个元素了。由于用vector装的，但是也是满足线性的。

　　二分：O(log n*m)

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) 
    {
        int n=matrix.size(), m=matrix[0].size();
        int L=0, R=n*m-1;
        while(L<R)
        {
            int mid=L+(R-L+1)/2;
            if(matrix[mid/m][mid%m]<=target)    L=mid;
            else    R=mid-1;
        }
        return matrix[L/m][L%m]==target;
    }
};

　　

　　迭代：O(n+m)

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) 
    {
        int n=matrix.size(), m=matrix[0].size();
        int i=0, j=m-1;
        while(i<n && j>=0)
        {
            if(target==matrix[i][j])    return true;
            if(target>matrix[i][j])    i++;
            else    j--;
        }
        return false;
    }
};