[74] Search a 2D Matrix

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),通过下图比较,时间复杂度明显优于之前的方案。
O(M+N)

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;  // 查找后没有找到
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值