【JZOF】二維陣列中的查詢

阿爾法貓發表於2018-08-13
陣列

題目：在一個二維陣列中，每一行都按照從左到右遞增的順序排序，每一列都按照從上到下遞增的順序。請完成一個函式，輸入這樣的一個二維陣列和一個整數，判斷陣列中是否含有該整數。

規律：首先先獲取陣列右上角的數字。如果該數字等於要查詢的數字，查詢過程結束；

如果該數字大於要查詢的數字，剔除這一列，因為該列的所有數字都比它大，向左移動一列，如果該數字小於要查詢的數字，剔除這一行，因為該行的所有數字都比它小，向下移動一行。

public class MatrixSearch{


       public static boolean find(int[][] matrix ,int number){
               //考慮程式的魯棒性，得要處理好可能的輸入異常
               if(matrix==null&&matrix.length<1&&matrix[0].length<1){
                      //如果該二維陣列沒有值
                      return false;
               }
               int rows=matrix.length,cols=matrix.length,row=0,col=cols-1;
               
               while(row<rows&&col>=0){
                    if(matrix[row][col]==number){
                           return true;
                    }else if(matrix[row][col]<number){
                           row++;
                    }else{
                         col--;
                    }
               }
               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));     // 健壯性測試，輸入空指標
複製程式碼

時間複雜度：O(m+n)


相關文章