面试题04:二维数组中的查找(C++)

题目链接https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof/

题目描述

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

题目示例

现有矩阵 matrix 如下:

[
[1, 4, 7, 11, 15],
[2, 5, 8, 12, 19],
[3, 6, 9, 16, 22],
[10, 13, 14, 17, 24],
[18, 21, 23, 26, 30]
]
给定 target = 5,返回 true。

给定 target = 20,返回 false。

解题思路

分析题目可以发现,数组元素之间存在一种特殊关系,即从左至右,从上至下都是递增的,因此我们选择以矩阵的右上角作为起点,并与target相比较,小于target则向下遍历,大于target则向左遍历,遍历完还没有则返回false。当然,我们也可以选择矩阵左下角为起点,与target相比较,大于target,则向上遍历,小于target,则向右遍历。

程序源码

class Solution {
public:
    bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
        //方法1:暴力法
        if(matrix.empty()) return false;
        int row = matrix.size(), col = matrix[0].size();
        for(int i = 0; i < row; i++)
        {
            for(int j = 0; j < col; j++)
            {
                if(matrix[i][j] == target) return true;
            }
        }
        return false;
        /*方法2:矩阵右上角开始
        if(matrix.empty()) return false;
        int row = 0, col = matrix[0].size() - 1;//右上角开始
        while(row < matrix.size() && col >= 0)
        {
            if(matrix[row][col] > target)
                col--;
            else if(matrix[row][col] < target)
                row++;
            else
                return true;
        }
        return false;
        */
        /*方法3:矩阵左下角开始
        if(matrix.empty()) return false;
        int row = matrix.size() - 1, col = 0;
        while(row >= 0 && col < matrix[row].size())
        {
            if(matrix[row][col] > target)
                row--;
            else if(matrix[row][col] < target)
                col++;
            else
                return true;
        }
        return false;
        */

    }
};

猜你喜欢

转载自www.cnblogs.com/wzw0625/p/12502045.html
今日推荐