JZ1 二维数组中的查找

    科技2024-01-02  98

    题目描述

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

    题目链接:https://www.nowcoder.com/practice/abc3fe2ce8e146608e868a70efebf62e?tpId=13&&tqId=11154&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

    题解:

    最直观的解法是双重循环遍历数组查找,但这样的时间复杂度是O(n^2),这样的解法是无法接受的;注意本题自带的递增规律,这就说明,对于在二维数组中的某一个数a(i, j),同一列下,行数下标大于i的都大于a,同理,同一行下,列数下标大于j的都大于a;按照这个规律,我们可以从最左上角的数开始查找,若左上角的数小于目标target,说明当前行没有比target大的,行下标向下移动,若左上角的数大于目标target, 说明当前列没有比target小的,列下标向左移动;用这种解法,无需遍历整个数组,一次排除一行或者一列,因此时间复杂度是O(n)。

    代码:

    class Solution { public:     bool Find(int target, vector<vector<int> > array) {         int size_row =  array.size();         int size_rol = array[0].size();         int index_row = 0, index_rol = size_rol - 1;         while(index_row != size_row && index_rol != -1){             if (array[index_row][index_rol] == target){                 return true;             }             if (array[index_row][index_rol] > target){                 index_rol --;             }             else if (array[index_row][index_rol] < target) {                 index_row ++;             }         }         return false;     } };

     

    Processed: 0.009, SQL: 8