LintCode: 28. 搜索二维矩阵 Java

题目大意

用O(logn)搜索m*n矩阵中的值。这个矩阵具有一下特性：每行中的整数从左到右是排序的；每行 第一个数大于上一行的最后一个整数。

解题思路

看到搜索，最先想到用二分法。此题可简单将二维看成一维，直接用二分法搜索，将index计算成行纵坐标即可。第二种方法：先二分搜索所在行，再二分搜索所在列。

public boolean searchMatrix(int[][] matrix, int target) {
        if (matrix == null || matrix.length==0)
            return false;
        int m=matrix.length, n=matrix[0].length, s=0, e=m*n;

        if (e == 1 && matrix[0][0] == target)
            return true;
        else if (e == 1 && matrix[0][0] != target)
            return false;
        while (s+1 < e) {
            int mid = (s+e)/2;
            int i=(mid%n==0 ? mid/n-1 : mid/n), j=(mid%n==0 ? n-1 : mid%n-1);
            if (matrix[i][j] == target) {
                return true;
            } else if (matrix[i][j] > target) {
                e = mid;
            } else if (matrix[i][j] < target){
                s = mid;
            } 
        }

        //头尾处理
        int i=(s%n==0 ? s/n-1 : s/n), j=(s%n==0 ? n-1 : s%n-1);
        if (matrix[i][j] == target)
            return true;
        i=(e%n==0 ? e/n-1 : e/n);
        j=(e%n==0 ? n-1 : e%n-1);
        if (matrix[i][j] == target)
            return true;

        return false;
    }