剑指offer-二维数组中的查找 (c++)

剑指offer-二维数组中的查找 (c++)

1. 题目描述

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

2. 解题思路

  1. 获取矩阵的行数和列数。
  2. 从第一行的最后一个数开始比较,如果是要查找的数,返回True;如果小于要查找的数,同一行向左查找,列数减少,col--;如果大于要查找的数,行数增加,row++。
  3. 依次遍历,直到出现row大于矩阵行数或者col<0时终止循环。

3.代码

#include <vector>
#include "iostream"

using namespace std;

class Solution {
public:
    bool Find(int target, vector<vector<int>> array) {
        // 获取矩阵的行数和列数
        int row = 0;
        int col = array[0].size() - 1;

        while (row < array.size() && col >= 0) {
            if (array[row][col] == target) {
                return true;
            } else if (array[row][col] < target) {
                // 查找下一行
                row++;
            } else {
                // 同一行内向左查找
                col--;
            }
        }
        return false;
    }
};

猜你喜欢

转载自www.cnblogs.com/Elijah-Z/p/12895928.html