判斷二維數組中是否包含某個數
題目:在一個二維數組中(每個一維數組的長度相同),每一行都按照從左到右遞增的順序排序,每一列都按照從上到下遞增的順序排序。請完成一個函數,輸入這樣的一個二維數組和一個整數,判斷數組中是否含有該整數。
總思路:如果要查找的數字不在數組的【右上角】,則每一次都刪除一列或一行,這樣每一步都可以縮小查找的范圍,直到找到要查找的數字,或則查找范圍為空。
具體步驟:
(1) 先取該二維數組最【右上角】的數字:
<1>如果該數 == 要查找的數,則直接返回true;
<2> 如果該數 > 要查找的數,則【刪除該列】;
<3> 如果該數 < 要查找的數,則【刪除該行】;
(2) 這樣每一步都可以縮小查找的范圍,直到找到要查找的數字,或則查找范圍為空。
(牛客網通過)
public class Demo01 {
public static void main(String[] args){
int[][] arr = {
{1,2,8,9},
{2,4,9,12},
{4,7,10,13},
{6,8,11,15}
};
int target = 6;
// int[][] arr={{}};
// int target = 0;
System.out.println(Find(target, arr));
}
public static boolean Find(int target, int [][] array) {
int row_len = array.length;
int col_len = array[0].length;
//如果二維數組為空
if (isEmpty(array))
return false;
//如果左上角小或比右下角要大,則一定不在其中
if (target < array[0][0] || target > array[row_len-1][col_len-1])
return false;
int i = 0, j = col_len -1;
if (!isEmpty(array))
{
// 1. 縮小【列】范圍
while (j > 0)
{
if (array[0][j] == target)
return true;
else if (array[0][j] > target)
j--;
else //即如果 array[0][j] < target
break;
}
// 2. 在找到的列范圍的基礎上, 縮小【行】范圍
while (i < row_len)
{
if (array[i][j] == target)
return true;
else if (array[i][j] < target)
i++;
else
break;
}
// 3.在最后剩下的范圍內尋找(在最后小范圍內暴力找)
int m,n;
for (m = i; m < row_len; m++)
for (n = 0; n < j; n++)
if (array[m][n] == target)
return true;
}
return false;
}
// 判斷二維數組是否為空
public static boolean isEmpty(int[][] array){
return (array == null || array.length == 0 || array[0].length == 0);
}
}