剑指offer 二维数组中的查找 C++

news2024/9/29 9:29:29

前言

一、题目

二、解题思路

1.直接查找

2.二分法

三、输出结果

前言

最近在牛客网刷题，刷到二维数组的查找，在这里记录一下做题过程

一、题目

描述
在一个二维数组中（每个一维数组的长度相同），每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。
[
[1,2,8,9],
[2,4,9,12],
[4,7,10,13],
[6,8,11,15]
]
给定 target = 7，返回 true。

给定 target = 3，返回 false。

二、解题思路

1.直接查找

看到题目的时候第一时间是直接遍历整个二维数组查找，这个方法比较简单，但是时间复杂度比较高，最坏的情况下要遍历数组里的所有数据。
代码如下：

#include<iostream>
#include<math.h>
#include<vector>
using namespace std;


class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        // 判断数组是否为空
        int m = array.size();
        if (m == 0) return false;

        int n = array[0].size();
        if (n == 0) return false;

        int r = 0, c = n - 1; // 右上角元素
        while (r < m && c >= 0) {
            if (target == array[r][c]) {
                return true;
            }
            else if (target > array[r][c]) {
                r++;
            }
            else {
                c--;
            }
        }
        return false;
    }
};

int main()
{
    vector<vector<int> > array = { {1,2,3,4,5,6,7,8,9},{11,12,13,14,15,16,17,18,19},{20,21,22,23,24,25,26,27,28,29} };
    Solution test;
    bool result = test.Find(6, array);

    std::cout << result << " " << std::endl;
    return 0;
}