这个算法的最坏时间复杂度是O(n*logm)。当然如果你对行也进行二分搜索,确定一个搜索的行的范围的话,时间复杂度会更低。
class Solution { public: bool searchMatrix(vector<vector<int> >& matrix, int target) { int n=matrix.size(),m=matrix[0].size(); for(int i=0;i<n;i++) { if(matrix[i][0]<=target&&matrix[i][m-1]>=target) { int left=0,right=m-1; while(left<=right) { int mid=(left+right)/2; if(matrix[i][mid]==target) return true; else if(matrix[i][mid]>target) right=mid-1; else left=mid+1; } } } return false; } };第二种方案:很巧妙的一个答案,我也是看了网上的题解才发现这种精巧的算法。算法主要是根据矩阵的特点得来的,如果我们从矩阵的右上角开始,如果target>cur时,那么行数当前行排除,行数+1;如果target<cur时,那么当前列排除,列数-1.遇到边界则结束,返回false;
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(matrix[i][j]==target) return true; else if(matrix[i][j]<target) i++; else j--; } return false; } };
版权声明:本文为博主原创文章,未经博主允许不得转载。
[leetcode] 240 Search a 2D Matrix II
原文地址:http://blog.csdn.net/nk_test/article/details/47778701