Search a 2D Matrix II

Search a 2D Matrix II

题意解释

大概意思就是采用一种优化算法从M*N行列都从小到大排序好的矩阵中找出目标数。

分治的思想

• 采用分治的思想。以题目给出矩阵为例，查找数字5。仔细观察矩阵，最右上角的数字为15，由于矩阵是列递增，所以数字5不可能在最右侧15这一列，我们便可将这一列不予考虑，将范围缩减了一列。
  　　
  　　　　　　　　　　　　　[1, 　　4, 　　7, 　11]
  　　　　　　　　　　　　　[2, 　　5, 　　8, 　12]
  　　　　　　　　　　　　　[3, 　　6, 　　9, 　16]
  　　　　　　　　　　　　　[10, 　13, 　14, 　17]
  　　　　　　　　　　　　　[18, 　21, 　23, 　26]

• 再判断数字11，同样11>5 11>5，又缩减一列。数字７同样小于５，在缩减一列，那么现在的矩阵变为：
  　　 　　　　　　　　　　　[1, 　　4,]
  　　　　　　　　　　　　　[2, 　　5]
  　　　　　　　　　　　　　[3, 　　6]
  　　　　　　　　　　　　　[10, 　13]
  　　　　　　　　　　　　　[18, 　21]

• 判断数字４时，由于5>45>4，目标值肯定不在4所在的行，去点这一行，在进行判断。
  　　　　　　　　　　　　　[2, 　　5]
  　　　　　　　　　　　　　[3, 　　6]
  　　　　　　　　　　　　　[10, 　13]
  　　　　　　　　　　　　　[18, 　21]

• 再判断数字5，找到目标值返回。
  　
  　这种算法时间复杂度为O(n), 代码如下：

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

auto gucciGang = []() {std::ios::sync_with_stdio(false);cin.tie(nullptr);cout.tie(nullptr);return 0;}();