有序矩阵查找问题

    给一个行和列都排好序的n*m矩阵,判断一个数是否存在于矩阵中,要求时间复杂度O(n+m),空间复杂度O(1)。

    遍历肯定太慢了,看到在矩阵中找,就联想到利用有序矩阵的性质一次否决掉一批元素。因为行和列都是有序的,所以对于矩阵中的每一个元素,同一行左面/同一列上面所有的元素都小于等于它,同一列下面/同一行右面所有的元素都大于等于它,所以每次把一个元素x和给定值a进行比较,都可以否决掉x同一行或者同一列一个方向上所有的元素,然后移动指针位置,把下一个元素拿来比较。

    如果在一个时刻拿来比较的x不能否决掉被否决行列上所有的剩余元素,这个算法的就没法单调向一个方向行进。如果x在剩下的矩阵的边上,那么每次都可以否决掉被否决行列上所有的剩余元素,x和a比较时,有相等、大于、小于三种可能,相等就返回true不用多说,在查找时要兼顾大于小于的情况,所以,选取右上角或者左下角,每次比较的时候,这一行/列上所有其他元素都位于x的一侧,所以每次都能,算法可以单调行进。

    这样解决思路就有了。以左上角为例,每次比较,如果x大于a,x下面所有元素排除,指针向左走一格;小于a,x左侧所有元素排除,指针向右走一格。每次比较之后,矩阵的剩余部分都还是一个矩阵,而指针指向剩下的矩阵左上角的元素。直到找到元素返回true,或者走不动了返回false。

    

	public boolean isInMatrix(int[][] m, int n) {
		int row = 0;
		int col = m[0].length-1;
		while(row <= m.length && col <= 0) {
			if(m[row][col] == n) {
				return true;
			}else if(m[row][col] < n) {
				row++;
			}else {
				col--;
			}
		}
		return false;
	}

猜你喜欢

转载自blog.csdn.net/daniel_2046/article/details/80784973