AcWing 15. 二维数组中的查找(模拟)

AcWing 15. 二维数组中的查找

题意

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

分析

1.仔细观察一下，发现最右边的那列数是比较特殊的，
这列数的所有元素都是所在行的最大值。
2.因此可以以右上角元素作为初始状态开始遍历，

2.1 由于每一列是从上到下递增，如果target小于当前元素，那么target肯定会小于位于当前元素下面的值；
2.2 由于每一行是从左到右递增，如果target大于当前元素，那么target肯定会大于位于当前元素左边的值；
2.3 如果当前元素等于target，直接返回true

代码

class Solution {
public:
    bool searchArray(vector<vector<int>> array, int target) {
        if (array.empty() || array[0].empty()) return false;
        //m为行的最小下标,n为列的最大下标
        int m = 0, n = array[0].size() - 1;
        
        while (m < array.size() && n >= 0) {
            int t = array[m][n];
            //如果当前元素等于target
            if (t == target) return true;
            //如果当前元素大于target,搜索方向往左移
            else if (t > target) n--;
            //如果当前元素小于target,搜索方向往下移
            else if (t < target) m++;
        }
        return false;
    }
};