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

在一个 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;}
};

相关内容

热门资讯

a股有几家上市公司 中国a股上... 齐鲁晚报讯祁鲁镇记者张通讯员6月15日,中国证监会山东监管局连续发布3份辅导工作总结报告,其中2份的...
开办小型日化厂违法吗 小型洗洁... 食品安全一直受到关注,尤其是在这样的大热天,人们对进口的东西更加谨慎。抚顺的朋友们,你们可能已经注意...
马云预言未来十大行业 暴利行业... 说到马云,很多人都很熟悉他。马云通过自己的努力和奋斗,逐渐开创了网上支付,他的眼光和勇气是非常独特的...
总是很自卑怎么办?做什么都做不... 总是很自卑怎么办?做什么都做不好,心态也很不好,感觉真的好累自卑跟性格有很大关系!你可能有些内向!其...
求一部国外电影,关于博物馆的故... 求一部国外电影,关于博物馆的故事讲的是在一个博物馆里每到晚上里面的东西就会复活,好像恐龙骨架什么的,...
我愿做江州司马为你泪湿青衫 我愿做江州司马为你泪湿青衫《琵琶行》白居易 座中泣下谁最多,江州司马青衫湿。同是天涯沦落人,相逢...
肩周炎怎么治? 肩周炎怎么治?肩周炎又称冻结肩、五十肩,是肩关节周围软组织慢性炎症性病变,主要以保守治疗为主。具体治...
谁有古代言而无信的例子? 谁有古代言而无信的例子?谁有古代言而无信的例子? 秦惠文王、张仪:秦惠文王更元十二年(前313年...
学生成长记录手册上家长寄语怎么... 学生成长记录手册上家长寄语怎么写可以写的抒情点,如“在生活中的每一天,都是一个阳光明媚的日子,因为,...
和伴侣在一起时会做哪些事情? 和伴侣在一起时会做哪些事情?情侣刚在一起的时候总是想把自己好的一面展现给对方,但是随着时间长了,感情...