欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

在行列都排好序的矩阵中找数

程序员文章站 2023-12-27 08:12:03
...

在行列都排好序的矩阵中找数

1、题目描述

  • 给定一个有N*M的整型矩阵matrix和一个整数K,matrix的每一行和每一 列都是排好序的。实现一个函数,判断K是否在matrix中。

  • 【要求】 时间复杂度为O(N+M),额外空间复杂度为O(1)。

2、思路。

  • 这里从数据状况来思考。由于数据每一行和列都是排好序的。

  • 每一次先判断右上角的数a,如果是小于就说明a下面的数不可能。

    ​ 便往左移动,到达b的位置。再做相同的判断。
    如果是大于,说明a左边的数不可能,便往下移动。依次类推

3、举例

从一个二维数组中查看是否包含0这个数

在行列都排好序的矩阵中找数

4、代码实现(java)

public static boolean isContains(int[][] matrix, int K) {
		int row = 0;
		int col = matrix[0].length - 1;
		while (row < matrix.length && col > -1) {
			if (matrix[row][col] == K) {
				return true;
			} else if (matrix[row][col] > K) {
				col--;
			} else {
				row++;
			}
		}
		return false;
	}
相关标签: 算法学习 算法

上一篇:

下一篇: