給定如下的n*n的數(shù)字矩陣,每行從左到右是嚴(yán)格遞增, 每列的數(shù)據(jù)也是嚴(yán)格遞增
1 2 3
3 5 6
4 8 9
現(xiàn)在要求設(shè)計(jì)一個(gè)算法, 給定一個(gè)數(shù)k 判斷出k是否在這個(gè)矩陣中。
算法設(shè)計(jì):
首先檢查左下角,如果這個(gè)數(shù)小于k,就說明這一列都小于k,接下來掃描右邊一列。
如果這個(gè)數(shù)大于k,就說明這一行都大于k,接下來掃描上面一行。
時(shí)間復(fù)雜度為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);
}
}