题目:
行和列都排好序的矩阵,请设计一个高效算法,快速查找矩阵中是否含有值x。
给定一个int矩阵mat,同时给定矩阵大小nxm及待查找的数x,请返回一个bool值,代表矩阵中是否存在x。
所有矩阵中数字及x均为int范围内整数。保证n和m均小于等于1000。
思路:判断矩阵右上角元素与待查找元素的大小,利用矩阵行列都有序的特性, 每次取出一行或一列元素。
算法时间复杂度 O(M+N)
贴代码:
bool findX(vector<vector<int> > mat, int n, int m, int x)
{
if (n < 0 || m < 0)
{
return false;
}
int row = 0, col = m - 1;
while (row < n && col >= 0)
{
// 右上角元素 = x 找到 返回true
if (mat[row][col] == x)
{
return true;
}
// 右上角元素 > x 最后一列元素不可能
else if(mat[row][col] > x)
{
col--;
}
// 右上角元素 < x 第一行元素不可能
else// if(mat[row][col] < x)
{
row++;
}
}
return false;
}