n*m的矩阵,行和列都递增有序,求是否出现target元素(面试题)
题目描述:给定一个n*m的矩阵,矩阵的每一行都是递增的,每一列也是递增的,给定一个元素target,问该target是否在矩阵中出现。
思路:先从最左下角的元素开始找,三种情况:
1. 如果该元素大于target,说明target一定不会出现在最后一行中(因为最后一行也是递增的,其他的更大于target)
2. 如果该元素小于target,说明target可能会出现在这一列的后面列中,因为最后元素是当前列当中最大的元素,前面的元素更小于target,所以只需要找该行的下一列
3. 如果该元素等于target,直接返回true
代码如下
bool searchMatrix(vector<vector<int> >& matrix, int target) { int n = matrix.size(); if (n == 0) return false; int m = matrix[0].size(); int row = n - 1; int col = 0; while (row >= 0 && col < m) { if (matrix[row][col] > target) row--; else if (matrix[row][col] < target) col++; else return true; } return false; }