问题描述 :
编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。要求使用二分查找。
该矩阵具有如下特性:
每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。
示例 1:
输入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 3
输出: true
示例 2:
输入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 13
输出: false
思路:
将二维矩阵逻辑上转换为以为数组(并不是真的建立一个数组),用idm、idn来记录矩阵里的行和列
#include<algorithm>
#include<iostream>
#include<math.h>
#include<stack>
#include<queue>
#include<cstring>
#include<vector>
using namespace std;
class Solution{
public:
bool searchMatrix(vector<vector<int> > matrix,int target){
int m,n,id,left,right;
n=matrix[0].size();
m=matrix.size();
if(m==0) return false;
right=m*n-1;
left=0;
while(left<=right){
id=(right+left)/2;
int idm=id/n;
int idn=id%n;
if(matrix[idm][idn]==target)
return true;
else{
if(target<matrix[idm][idn])
right=id-1;
else
left=id+1;
}
}
return false;
}
};
int main()
{
vector<vector<int> > matrix;
int target;
int m,n,e;
cin>>m>>n;
for(int i=0; i<m; i++)
{
vector<int> aRow;
for(int j=0; j<n; j++)
{
cin>>e;
aRow.push_back(e);
}
matrix.push_back(aRow);
}
cin>>target;
bool res=Solution().searchMatrix(matrix,target);
cout<<(res?"true":"false")<<endl;
return 0;
}