【题目】 给定一个有N*M的整型矩阵matrix和一个整数K,matrix的每一行和每一 列都是排好序的。实现一个函数,判断K是否在matrix中。 例如:
0 1 2 5
2 3 4 7
4 4 4 8
5 7 7 9
如果K为7,返回true;如果K为6,返回false。
【要求】 时间复杂度为O(N+M),额外空间复杂度为O(1)
考虑到都为有序那么可以制定规则,而不是暴力搜索
import java.util.*;
public class Main {
/**
* @param args
*/
public static void main(String[] args) {
// TODO Auto-generated method stub
//设定行数和列数
int[][] matrix = new int[][] { { 0, 1, 2, 3, 4, 5, 6 },// 0
{ 10, 12, 13, 15, 16, 17, 18 },// 1
{ 23, 24, 25, 26, 27, 28, 29 },// 2
{ 44, 45, 46, 47, 48, 49, 50 },// 3
{ 65, 66, 67, 68, 69, 70, 71 },// 4
{ 96, 97, 98, 99, 100, 111, 122 },// 5
{ 166, 176, 186, 187, 190, 195, 200 },// 6
{ 233, 243, 321, 341, 356, 370, 380 } // 7
};
int K = 233;
System.out.println(isContains(matrix, K));
}
public static boolean isContains(int[][] arr,int k){
int endR =arr.length-1;
int endL =arr[0].length-1;
int x=0,y=endL;
while(x<=endR&&y>-1){
if(arr[x][y]>k){//由于是排好序的所以如果现在的值比要找的值大,
//那么向左找有可能找到对应的值
y--;
}else if(arr[x][y]<k){//由于是排好序的所以如果现在的值比要找的值小,
//那么向下找有可能找到对应的值
x++;
}else{
return true;
}
}
return false;
}
}