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

来源:互联网 发布:淘宝买苹果主板靠谱吗 编辑:程序博客网 时间:2024/06/08 08:21

题目:

行和列都排好序的矩阵,请设计一个高效算法,快速查找矩阵中是否含有值x。

给定一个int矩阵mat,同时给定矩阵大小nxm及待查找的数x,请返回一个bool值,代表矩阵中是否存在x。

所有矩阵中数字及x均为int范围内整数。保证nm均小于等于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 找到 返回trueif (mat[row][col] == x){return true;}// 右上角元素 > x 最后一列元素不可能else if(mat[row][col] > x){col--;}// 右上角元素 < x 第一行元素不可能else// if(mat[row][col] < x){row++;}}return false;}

0 0