LintCode 搜索二维矩阵
写出一个高效的算法来搜索 m × n矩阵中的值。
这个矩阵具有以下特性:
- 每行中的整数从左到右是排序的。
- 每行的第一个数大于上一行的最后一个整数
样例
考虑下列矩阵:
[
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
给出 target = 3,返回 true
剑指offer 面试题目3 跟这个相仿 :找右上角或者左下角 可以排除一行或一列 效率高
这个题烦死了 先开始没考虑鲁棒性 后来有数组下表越界
最后总算通过了。。。
class Solution {
public:
/**
* @param matrix, a list of lists of integers
* @param target, an integer
* @return a boolean, indicate whether matrix contains target
*/
bool searchMatrix(vector<vector<int> > &matrix, int target) {
// write your code here
int found=false;
if(matrix.empty()|| matrix[0].empty())
return found;
if (target < matrix[0][0] || target > matrix.back().back())
return found;
int rows=matrix.size();
int columns=matrix[0].size();
int row=0;
int column=columns-1;
while(row < rows && column>=0)
{
if(matrix[row*columns][column]==target)
{ found=true;
break;
}
else if(matrix[row*columns][column]>target)
--column;
else
++row;
}
return found;
}
};

浙公网安备 33010602011771号