走自己的路----240. 搜索二維矩陣 II

編寫一個高效的算法來搜索 m x n 矩陣 matrix 中的一個目標值 target。該矩陣具有以下特性：

每行的元素從左到右升序排列。
每列的元素從上到下升序排列。
示例:

現有矩陣 matrix 如下：

[
  [1,   4,  7, 11, 15],
  [2,   5,  8, 12, 19],
  [3,   6,  9, 16, 22],
  [10, 13, 14, 17, 24],
  [18, 21, 23, 26, 30]
]
給定 target = 5，返回 true。

給定 target = 20，返回 false。

 

解法1：沿對角線二分查找

public boolean searchMatrix(int[][] matrix, int target) {
        if(matrix.length==0) return false;
        int l = Math.min(matrix.length,matrix[0].length);
        for(int i=0;i<l;i++){
            boolean rowSearch = binarySearch(matrix,i,true,target);
            boolean colSearch = binarySearch(matrix,i,false,target);
            if(rowSearch||colSearch)
                return true;
        }
        return false;
    }
    public boolean binarySearch(int[][] matrix,int i,boolean flag,int target){

        int start = i;
        int end = flag?matrix[0].length-1:matrix.length-1;
        while(start<=end){
            int mid = (start+end)/2;
            if(flag){
                if(matrix[i][mid]>target){
                    end = mid-1;
                }else if(matrix[i][mid]<target){
                    start = mid +1;
                }else{
                    return true;
                }
            }else{
                if(matrix[mid][i]>target){
                    end = mid-1;
                }else if(matrix[mid][i]<target){
                    start = mid +1;
                }else{
                    return true;
                }
            }
        }
        return false;
    }

解法二：右上角開始查找

public boolean searchMatrix(int[][] matrix, int target) {
        if(matrix.length==0) return false;
        int rows = matrix.length;
        int cols = matrix[0].length;
        int i = 0,j=cols-1;
        while(i<rows&&j>=0){
            if(matrix[i][j]>target){
                j--;
            }else if(matrix[i][j]<target){
                i++;
            }else{
                return true;
            }
        }
        return false;
    }