二维数组中的查找

Posted 2021-12-09 clarencezzh

在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。

请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

样例：

输入数组：

[
  [1,2,8,9]，
  [2,4,9,12]，
  [4,7,10,13]，
  [6,8,11,15]
]

如果输入查找数值为7，则返回true，

如果输入查找数值为5，则返回false。

---

首先有两种思路：

二分查找是最直观的，横的方向和竖方向都是有序数列。所以可以遍历横方向然后利用二分查找。O(nlogn)

右上角往下是增加，往左走是减小。所以利用这个规律查找目标值。O(n)

bool searchArray(vector<vector<int>> array, int target) {
    if (array.empty() || array[0].empty()) return false;
    int cols=array.at(0).size()-1;
    int rows=0;
    while(cols>=0&&rows<array.size()){
        if(array[rows][cols]==target)
            return 1;
        else if(array[rows][cols]<target){
            rows++;
        }else{
            cols--;
        }

    }
    return 0;
}