給定如下的n*n的數字矩陣,每行從左到右是嚴格遞增, 每列的數據也是嚴格遞增
1 2 3
3 5 6
4 8 9
現在要求設計一個算法, 給定一個數k 判斷出k是否在這個矩陣中。
算法設計:
首先檢查左下角,如果這個數小于k,就說明這一列都小于k,接下來掃描右邊一列。
如果這個數大于k,就說明這一行都大于k,接下來掃描上面一行。
時間復雜度為O(n+m-1)
public class JuzhenN {
public static void main(String[] args){
int[][] data = {
{1, 2, 3, 4, 5},
{2, 3, 5, 6, 7},
{6, 8, 9,10,11},
{7, 9,10,12,13},
{8,10,11,15,14}
};
int k = 13;
searchK(data, k);
}
private static void searchK(int[][] data, int k) {
int m = data.length;
int n = data[0].length;
for(int i = m - 1, j = 0; i >= 0 && j < n; ){
if(data[i][j] == k){
System.out.println(true);
return;
}else if(data[i][j] < k){
j++;
}else{
i--;
}
}
System.out.println(false);
}
}