【劍指offer】二維陣列中的查詢

Lrixin發表於2021-01-20

題目連結:https://www.nowcoder.com/practice/abc3fe2ce8e146608e868a70efebf62e?tpId=13&&tqId=11154&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking
題目描述
在一個二維陣列中(每個一維陣列的長度相同),每一行都按照從左到右遞增的順序排序,每一列都按照從上到下遞增的順序排序。請完成一個函式,輸入這樣的一個二維陣列和一個整數,判斷陣列中是否含有該整數。
示例1
輸入

7,[[1,2,8,9],[2,4,9,12],[4,7,10,13],[6,8,11,15]]

返回值

true

樸素演算法:搜尋整個二維陣列,時間複雜度為O(N2);

public class Solution {
    public boolean Find(int target, int [][] array) {
        int n=array.length;
        int m=array[0].length;
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                if(array[i][j]==target){
                    return true;
                }
                /*if(array[i][j]>target){
                    break;//這也是一個改進辦法,不過提升效率不高
                }*/
            }
        }
        return false;
    }
}

二分演算法:因為每一行都按照從左到右遞增的順序排序,每一列都按照從上到下遞增的順序排序。我們可以利用陣列的這個特點對二維陣列進行二分查詢。
Java Code

public class Solution {
    public boolean Find(int target, int [][] array) {
        if(array==null){
            return false;
        }
        int n=array.length;
        int m=array[0].length;
        if(n<=0||m<=0){
            return false;
        }
        int i=0,j=m-1;//最開始選取右上角的數字
        while(i<n&&j>=0){//退出條件
            if(target==array[i][j]){
                return true;//找到則退出
            }
            else if(target>array[i][j]){
                i++;//target在第i行的下面
            }
            else{
                j--;//target在第j列的左邊
            }
        }
        return false;//陣列不存在target
    }
}

Cpp Code

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int n=array.size();
        int m=array[0].size();
        if(n<=0||m<=0){
            return false;
        }
        int i=0,j=m-1;
         while(i<n&&j>=0){//退出條件
            if(target==array[i][j]){
                return true;//找到則退出
            }
            else if(target>array[i][j]){
                i++;//target在第i行的下面
            }
            else{
                j--;//target在第j列的左邊
            }
        }
        return false;//陣列不存在target
    }
};

相關文章