劍指offer1-二維數組中的查找
1、題目描述
在一個二維數組中(每個一維數組的長度相同),每一行都按照從左到右遞增的順序排序,每一列都按照從上到下遞增的順序排序。請完成一個函數,輸入這樣的一個二維數組和一個整數,判斷數組中是否含有該整數。
2、思路分析
該二維數組有個特性,由上到下遞增,由左到右遞增。總結規律我們可以發現,我們應該從矩陣的右上角或者左下角可是查詢。每一次可以刪除一行或者一列,進而縮小查找範圍。
以右上角爲例,首先選取右上角的數字,如果該數字等於要查找的數字,則查找過程結束;如果該數字大於要查找的數字,則說明該列其他元素都大於要查找的數字,便可以刪掉該列;如果該數字小於要查找的數字,則說明該行其他元素也都小於要查找的數字,便可以刪掉該行。這樣,每一次比較都可以剔除一行或者一列,進而縮小查找範圍,時間複雜度爲O(n)。。
3、代碼
package com.jianzhioffer;
public class offer_01 {
public static boolean find(int[][] matrix,int number) {
//判斷邊界
if(matrix==null || matrix.length<1 || matrix[0].length<1 ) {
return false;
}
//定義行數 列數
int rows=matrix.length;
int cols=matrix[0].length;
//從右上角開始查找
int row=0;
int col=cols-1;
//要查找的位置確保在數組內
while(row>=0 && row<rows && col>=0 && col<cols) {
//如果找到就直接退出
if(matrix[row][col]==number) {
return true;
} else if(matrix[row][col]>number) {
//如果找到的數比要找的數大,說明要找的數在當前數的左邊
col--;
} else {
row++;
}
}
return false;
}
public static void main(String[] args) {
int[][] matrix = {
{1, 2, 8, 9},
{2, 4, 9, 12},
{4, 7, 10, 13},
{6, 8, 11, 15}
};
System.out.println(find(matrix, 7)); // 要查找的數在數組中
System.out.println(find(matrix, 5)); // 要查找的數不在數組中
System.out.println(find(matrix, 1)); // 要查找的數是數組中最小的數字
System.out.println(find(matrix, 15)); // 要查找的數是數組中最大的數字
System.out.println(find(matrix, 0)); // 要查找的數比數組中最小的數字還小
System.out.println(find(matrix, 16)); // 要查找的數比數組中最大的數字還大
System.out.println(find(null, 16)); // 健壯性測試,輸入空指針
}
}