1. 题目描述
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.
题目给定条件为一行中的数据从左到右递增,且下一行的第一个数一定大于前一行的最后一个数
2. 解题思路
一开始的思路是从[0,0]开始沿列进行查找,找到最后一个小于target的数字所在行数i,再沿着i行进行查找,找到等于target的数字返回true,找到一个大于target的数字返回false,这样搜索次数最多为m+n次,时间复杂度为O(M+N),但之后都超时了,不知是不是我的code写的不对
后来想到根据题目给定的顺序,如果横向拉伸开的话就是一个已经从小到大排序好的一维数组,[1, 3, 5, 7, 10, 11, 16, 20, 23, 30, 34, 50],通过二分查找的最差时间复杂度为O(logM+logN),通过下图比较,时间复杂度明显优于之前的方案。
3. Code
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
int m(matrix.size()), n(matrix[0].size()); // m行n列
if(m*n == 0) // 数组元素个数为0时
return false;
int low(0), high(m*n-1);
while(low <= high){
int mid = low + (high - low)/2; // 二分查找的点
// 把坐标分解为matirx[x][y]
int x = mid/n;
int y = mid%n;
int midval = matrix[x][y]; // 中间变量的值
if(midval == target)
{
return true;
}
else if(midval > target) // 如果大了取前半段
{
high = mid-1;
}
else if(midval < target) // 如果小了取后半段
{
low = mid+1;
}
}
return false; // 查找后没有找到
}
};