劍指leetcode—搜索二維矩陣

題目描述:編寫一個高效的算法來判斷 m x n 矩陣中,是否存在一個目標值。該矩陣具有如下特性:

每行中的整數從左到右按升序排列。
每行的第一個整數大於前一行的最後一個整數。

示例 1:

輸入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 3
輸出: true

示例 2:

輸入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 13
輸出: false

方法一:

利用二維矩陣的結構特徵來進行二分搜索

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        if(matrix.length==0||matrix[0].length==0)
        return false;
        int m=matrix.length;
        int n=matrix[0].length;
        for(int i=0;i<m;i++)
        {
            if(matrix[i][n-1]>target)
            {
                int start=0;
                int end=n-1;
                while(start<=end)
                {
                    int mid=(start+end)/2;
                    if(matrix[i][mid]==target)
                    return true;
                    else if(matrix[i][mid]>target)
                    end=mid-1;
                    else
                    start=mid+1;
                }
                break;
            }
            else if(matrix[i][n-1]==target)
            return true;
            else
            continue;
        }
        return false;
    }
}

方法一的另一種形式,也可以不採用二分的辦法,這樣更簡潔,採用倒退迭代的思想

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
       if(matrix.length==0||matrix==null)
        return false;
        int m=matrix.length;
        int n=matrix[0].length;
        int row=0;
        int column=n-1;
        while(row<m&&column>=0)
        {
            if(matrix[row][column]>target)
            column--;
            else if(matrix[row][column]<target)
            row++;
            else
            return  true;
        }
        return false;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章