leetcode原题链接:搜索二维矩阵
题目描述
给你一个满足下述两条属性的 m x n
整数矩阵:
- 每行中的整数从左到右按非递减顺序排列。
- 每行的第一个整数大于前一行的最后一个整数。
给你一个整数 target
,如果 target
在矩阵中,返回 true
;否则,返回 false
。
示例 1:
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 3 输出:true
示例 2:
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 13 输出:false
提示:
m == matrix.length
n == matrix[i].length
1 <= m, n <= 100
-104 <= matrix[i][j], target <= 104
解题方法:二分查找。从右上角(x,y)开始查找,如果nums(x,y)等于target直接返回;若target < nums(x,y),y--,即向左边查找;若target>nums(x,y),则x++,则向下查找。
C++代码
#include <iostream>
#include <vector>
class Solution {
public:
bool searchMatrix(std::vector<std::vector<int>>& matrix, int target) {
int rnum = matrix.size(); // 行数
int cnum = matrix[0].size(); //列数
int x = 0;
int y = cnum - 1; // (x, y)从右上角开始查找
while ( x >= 0 && x < rnum && y >= 0 && y < cnum) {
if (target == matrix[x][y]) {
return true;
} else if (target < matrix[x][y]) { // 当前值大于目标值,则向左缩小查找空间
y = y - 1;
} else { // 当前值小于目标值,则向下扩大查找空间
x = x + 1;
}
}
return false;
}
};