Write an efficient algorithm that searches for a value in an m x n matrix. This matrix has the following properties:
- Integers in each row are sorted from left to right.
- The first integer of each row is greater than the last integer of the previous row.
For example,
Consider the following matrix:
[ [1, 3, 5, 7], [10, 11, 16, 20], [23, 30, 34, 50] ]
Given target = 3
, return true
.
从第0行末尾元素开始,当做中间元素,比他大的一定在下一行,小的一定在同一行。总的时间复杂度。O(MAX(M,N));
#include <iostream>
#include <string>
#include <vector>
using namespace std;
bool searchMatrix(vector<vector<int> > &matrix, int target);
int main()
{
int A[][4]={{1,3,5,7},{10,11,16,20},{23,30,34,50}};
vector<vector<int> > matrix(4,vector<int> (4));
for(int i = 0;i < matrix.size();i++)
{
for(int j = 0;j < matrix[0].size();j++)
{
matrix[i][j] = A[i][j];
}
}
int val;
cin>>val;
if(searchMatrix(matrix,val))
{
cout<<"true"<<endl;
}
else
{
cout<<"false"<<endl;
}
return 0;
}
bool searchMatrix(vector<vector<int> > &matrix, int target)
{
if(matrix.empty())return false;
int row = 0;
int cloumn = matrix[0].size()-1;//开始为第0行末尾元素
while(row < matrix.size() && cloumn >=0)
{
if(matrix[row][cloumn] == target)
{
return true;
}
else if(matrix[row][cloumn] > target)//同列查找
{
cloumn -- ;
}
else//下一行查找
{
row++;
}
}
return false;
}
思路二:对于此题可以想象把该矩阵拉伸成一维数组,因为此处有从小到大关系。进行定位是转换为二维坐标即可。时间复杂度O(lg(M*N));
bool searchMatrix(vector<vector<int> > &matrix, int target)
{
if(matrix.empty())return false;
int left_index = 0;
int right_index = matrix.size()*matrix[0].size()-1;
int mid = 0;
while(left_index <= right_index)
{
mid = (left_index + right_index)/2;
if(matrix[mid/matrix[0].size()][mid%matrix[0].size()] == target)//转换成二位坐标定位
{
return true;
}
else if(matrix[mid/matrix[0].size()][mid%matrix[0].size()] < target)//转换成二位坐标定位
{
left_index = mid + 1;
}
else
{
right_index = mid - 1;
}
}
return false;
}