题目链接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; */ } };