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

题目:

行和列都排好序的矩阵,请设计一个高效算法,快速查找矩阵中是否含有值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 找到 返回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;
	}

    原文作者:查找算法
    原文地址: https://blog.csdn.net/u013575812/article/details/50001837
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞