[数组]二维数组中的查找

118 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

问题

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

思路

从左下角开始遍历,target 大于左下角元素,往右侧寻找,target 小于左下角元素,向上寻找。

代码实现

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        // x 正方向为向下递增方向,y 正方向为向右递增方向
        int x = array.size() - 1;
        int y = 0;
        // 从左下角开始遍历,target 大于左下角元素,往右侧寻找,target 小于左下角元素,向上寻找
        while (x >= 0 && y < array[x].size())
        {
            if (array[x][y] < target) y++;
            else if (array[x][y] > target) x--;
            else return true;
        }
        return false;
    }
};