剑指 Offer 04. 二维数组中的查找
创始人
2024-06-03 15:00:16

在一个 n * m的二维数组中,每一行都按照从左到右 非递减 的顺序排序,每一列都按照从上到下 非递减 的顺序排序。请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。

示例:

现有矩阵 matrix 如下:

[[1,   4,  7, 11, 15],[2,   5,  8, 12, 19],[3,   6,  9, 16, 22],[10, 13, 14, 17, 24],[18, 21, 23, 26, 30]
]

给定 target = 5,返回true
给定 target = 20,返回false

思路:

解法一:

对每一行进行二分查找,复杂度O(nlogm)O(nlogm)O(nlogm)。

代码:

class Solution {
public:bool findNumberIn2DArray(vector>& matrix, int target) {if(matrix.size() == 0 || matrix[0].size() == 0) return false; //对矩阵判空int n = matrix.size(),  m = matrix[0].size();for(int i = 0; i < n; ++ i) {int l = lower_bound(matrix[i].begin(), matrix[i].end(), target) - matrix[i].begin();if(l < m && matrix[i][l] == target) return true;}return false;}
};

解法二:

利用矩阵matrix的性质从右上角(或左下角)开始进行搜索实际上从右上角开始看,我们会发现整个矩阵会变成一个类二叉搜索树的东西,这就很简单了 )。
这里我们以右上角为例,在每一步搜索过程中,如果我们位于位置(x,y),那么我们希望以matrix的左下角为当前搜索矩阵的左下角,以(x,y)当前搜索矩阵的右上角,形成一个新的搜索矩阵。在这个搜索矩阵中进行以下搜索,如果:

  • matrix[x,y] = target,说明搜索完成;
  • matrix[x,y] > target,由于每一列的元素都是非递减排列的,那么在当前的搜索矩阵中,所有位于第y列的元素都是严格大于target的,因此我们可以将它们全部忽略,即将y减少1
  • matrix[x,y] < target,同理,由于每一行的元素都是非递减排列的,那么在当前的搜索矩阵中,所有位于第x行的元素都是严格小于target的,因此我们可以将它们全部忽略,即将x增加1
  • 在搜索的过程中,如果我们超出了矩阵的边界,那么说明矩阵中不存在
    target

算法的时间复杂度为O(n+m)O(n + m)O(n+m),yyy最多减少mmm次,xxx最多增加nnn次,总搜索次数最大为n+mn + mn+m。

代码:

class Solution {
public:bool findNumberIn2DArray(vector>& matrix, int target) {if(matrix.size() == 0 || matrix[0].size() == 0) return false;//对矩阵判空int n = matrix.size(), m = matrix[0].size();int x = 0, y = m - 1;while(x < n && y >= 0) {if(matrix[x][y] < target) {x ++;} else if(matrix[x][y] > target) {y --;} else {return true;}}return false;}
};

相关内容

热门资讯

光稳定剂领域的“破局者” (来源:劳动午报)转自:劳动午报 在化工行业转型升级的浪潮中,有一位用技术之光照亮产业前路的巾帼先锋...
GDP超3.37万亿元 同比增... 1月26日,重庆市2025年“经济报表”出炉。总体来看,2025年全市各项政策协同发力,生产供给支撑...
亚洲室内田径锦标赛 将在天津举... (来源:劳动午报)转自:劳动午报 新华社电 记者26日从天津市体育局获悉,第十二届亚洲室内田径锦标赛...
着力保障和改善民生 提高群众... (来源:劳动午报)转自:劳动午报 本报讯(记者 余翠平 宗晓畅)1月26日上午,北京市政协十四届四次...
国际油气价格迎来阶段性反弹 转自:证券日报    本报记者 贺王娟    近期,有色金属价格持续走高,国际原油、天然气价格也迎来...