lintcode刷題之搜索二維矩陣,題目如下所示:
寫出一個高效的算法來搜索 m × n矩陣中的值。
這個矩陣具有以下特性:
- 每行中的整數從左到右是排序的。
- 每行的第一個數大於上一行的最後一個整數。
考慮下列矩陣:
[
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
給出 target = 3
,返回 true
1、由於每一個vector裏面都是有序的,所以首先找到待搜索元素在哪一個vector裏面;
2、找到待搜索元素所在的vector之後再對該行數據進行二分法查找;
具體的c++代碼如下:
class Solution {
public:
/**
* @param matrix, a list of lists of integers
* @param target, an integer
* @return a boolean, indicate whether matrix contains target
*/
bool erfenchazhao(vector<int>&v,int target)
{
if(v.size()==0)
{
return false;
}
int i,j;
i=0,j=v.size()-1;
int temp=(i+j)/2;
while(i<=j)
{
if(v[temp]<target)
{
i=temp+1;
temp=(i+j)/2;
}
else if(v[temp]>target)
{
j=temp-1;
temp=(i+j)/2;
}
else
{
return true;
}
}
return false;
}
bool searchMatrix(vector<vector<int> > &matrix, int target) {
// write your code here
int row=matrix.size();
if(row==0)
{
return false;
}
int col=matrix[0].size();
int t;
int i,j;
for(i=0;i<row;i++)
{
if(target<matrix[i][0])
{
t=i;
break;
}
}
if(i==row)
{
return erfenchazhao(matrix[row-1],target);
}
if(t==0)
{
return false;
}
else
{
return erfenchazhao(matrix[t-1],target);
}
}
};