28搜索二维矩阵

2017-10-26  本文已影响0人  shenlong77

写出一个高效的算法来搜索 m × n矩阵中的值。
这个矩阵具有以下特性:
每行中的整数从左到右是排序的。
每行的第一个数大于上一行的最后一个整数。

[
  [1, 3, 5, 7],
  [10, 11, 16, 20],
  [23, 30, 34, 50]
]

时间复杂度O(log(m+n))

public class Solution {
    /*
     * @param matrix: matrix, a list of lists of integers
     * @param target: An integer
     * @return: a boolean, indicate whether matrix contains target
     */
    public boolean searchMatrix(int[][] matrix, int target) {
        // write your code here
        //判断数组是否为null,是否为空,target是否不再数组里
        if(matrix==null||matrix.length==0||matrix[0][0]>target||matrix[matrix.length-1][matrix[0].length-1]<target){
            return false;
        }
        int m=matrix.length;
        int n=matrix[0].length;
        int startIndex=0;
        int endIndex=m*n-1;
        while(true){
            int middleIndex=(startIndex+endIndex)/2;
            int row=middleIndex/n;
            int col=middleIndex%n;
            if(matrix[row][col]==target){
                return true;
            }
            if(startIndex>=endIndex){
                return false;
            }else{
                if(target<matrix[row][col]){
                    endIndex=middleIndex-1;
                }else{
                    startIndex=middleIndex+1;
                }
            }
        }
    }
}
上一篇 下一篇

猜你喜欢

热点阅读