Java实现行列递增矩阵的查找
1 问题描述
在一个m行n列的二维数组中,每一行都按照从左到右递增的顺序排列,每一列都按照从上到下递增的顺序排列。现在输入这样的一个二维数组和一个整数,请完成一个函数,判断数组中是否含有该整数。
2 解决方案
2.1定位法
下面算法的时间复杂度为O(m + n),空间复杂度为O(1)。
package com.liuzhen.practice;
public class Main {
public boolean YoungMatrix(int[][] A, int key) {
int i = 0, j = A[0].length - 1;
int temp = A[i][j];
while(true) {
if(temp == key) {
return true;
} else if(temp < key && i < A.length - 1) {
temp = A[++i][j];
} else if(temp > key && j > 0) {
temp = A[i][--j];
} else {
return false;
}
}
}
public static void main(String[] args) {
Main test = new Main();
int[][] A = {{1,2,8,9},{2,4,9,12},{4,7,10,13},{6,8,11,15}};
if(test.YoungMatrix(A, 6))
System.out.println("矩阵A中包含元素6");
else
System.out.println("矩阵A中不包含元素6");
if(test.YoungMatrix(A, 5))
System.out.println("矩阵A中包含元素5");
else
System.out.println("矩阵A中不包含元素5");
}
}
运行结果:
矩阵A中包含元素6
矩阵A中不包含元素5