0
点赞
收藏
分享

微信扫一扫

搜索二维矩阵

elvinyang 2022-02-02 阅读 81

知识点:


 

我们怎么求二位矩阵第k个数的行数和列数。

我们这里矩阵从零开始,如图所示,我们标记为5的数,为第5个数

它的行数为5/m,列数为5%m,这里m表示列数,如图所以我们总共由4列,1,1

我们根据题意可知,如果我们把矩阵的数按顺序展开成1行的话将会是单调递增的,那么我们就可以用二分来做,这里我们可以把二维矩阵拆为1行,每次二分取第mid的数,通过转换成矩阵的数与target比较得出答案。

这里我们的下表总共到n*m-1,从0到n*m-1真好为n*m个数

步骤:

我们每次枚举的mid把它转换到二维矩阵的数,把他与target比较最终得到答案。

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& a, int target) {
      if(a.empty()||a[0].empty())return false;
      int n=a.size(),m=a[0].size();
      int l=0,r=n*m-1;
      while(l<r){
          int mid=(l+r)/2;
          if(a[mid/m][mid%m]>=target)r=mid;
          else l=mid+1;
      }
      return a[r/m][r%m]==target;
      }
};
举报

相关推荐

0 条评论