您的位置:首页 > 编程语言 > C语言/C++

有序矩阵查找的快速算法(C++版)

2015-11-23 22:08 281 查看
题目:

行和列都排好序的矩阵,请设计一个高效算法,快速查找矩阵中是否含有值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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: