Search a 2D Matrix

最新推荐文章于 2021-07-10 01:06:06 发布

原创最新推荐文章于 2021-07-10 01:06:06 发布 · 215 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种高效的算法，用于在一特殊类型的二维矩阵中搜索特定值。该矩阵的特性为每一行元素从左到右递增排序，且每行首个元素大于前一行的末尾元素。文章提供了一个具体示例，并给出了实现这一算法的C++代码。

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.

由于在二维数组中的所有元素都是已经排好序的，所以我们可以从第一行开始，用每行的第一个元素与目标进行比较，当出现某行的第一个元素大于目标时，则目标如果在数组中就只能在上一行的元素中，如果每一行的第一个元素都不大于目标，则目标只可能在最后一行中。

我们把所有行的第一个元素扫一遍就可以锁定目标只可能出现在某一行中，只要对这行的元素进行线性查找就能得出答案。这个算法的时间复杂度是O（m+n），源代码如下：

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int m=matrix.size(),n=matrix[0].size();
        int i=0;
        for(;i<m;i++)
        	if(matrix[i][0]>target) break;
        i--;
        if(i<0) return false;
        for(int j=0;j<n;j++)
        {
        	if(matrix[i][j]==target) return true;
        	else if(matrix[i][j]>target) return false;
        }
        return false;
    }
};