给定M×N矩阵,每一行、每一列都按升序排列,请编写代码找出某元素。
示例:
现有矩阵 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。
题解:这类问题,从左上角或者右下角开始进行处理,都会遇到向左/向下 或者向上/向右都是增加的问题。
因此可以从左下角开始,向上或者向右,分别是减少和增加,这样就进行了类似二分查找的动态规划。
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
if(matrix.empty()) return false;
int matrix_y = matrix.size();
int matrix_x = matrix[0].size();
int anchor_x = 0;
int anchor_y = matrix_y-1;
while(anchor_x<matrix_x && anchor_y>=0){
if(matrix[anchor_y][anchor_x]<target){
anchor_x++;
}else if(matrix[anchor_y][anchor_x]>target){
anchor_y--;
}else{
break;
}
}
if(anchor_x<matrix_x && anchor_y>=0) return true;
else return false;
}
};