二维数组中的查找

Posted 2021-03-24 ponda

题目描述

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

最开始题目理解错了，以为二维数组是从上到下，从左到右递增排列的。题目的意思是每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序，数组可能是这样的：[[1,2,8,9],[4,7,10,13]]。

错误代码

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int i;
        if(array.size()==0){
            return false;
        }
        if(array[0].size()==0){
            return false;
        }
        for (i = 0; i<array.size(); i++) {
            if (target<array[i][0]) {
                if (i == 0) {
                    return false;
                }
                int j;
                for (j = 0; j < array[i - 1].size(); j++) {
                    if (target == array[i - 1][j]) {
                        cout << i - 1 << ", " << j << ", " << array[i - 1][j] << endl;
                        return true;
                    }
                }
                return false;
            }
            else {
                if (i == array.size() - 1) {
                    int j;
                    for (j = 0; j<array[i].size(); j++) {
                        if (target == array[i - 1][j]) {
                            return true;
                        }
                    }
                    return false;
                }
            }
        }
        return false;
    }
};

这段代码在只满足从上到下，从左到右递增排列的二维数组。

正确代码：

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

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