刷题笔记(10)

Posted 2021-07-18 HardyDragon_CC

搜索二维矩阵

这里可以使用二分查找同时对行和列进行查找。选开始位置时选矩阵最短的一边，遍历最少次数。二分查找处理行和列的操作不同。

class Solution {
    private boolean binarySearch(int[][] matrix, int target, int start, boolean vertical) {
        int lo = start;
        // true hi为列的最大下标；false hi是行数的最大下标
        int hi = vertical ? matrix[0].length-1 : matrix.length-1;

        while (hi >= lo) {
            int mid = (lo + hi)/2;
            if (vertical) { // searching a column 查找列
            	// 横向的二分
                if (matrix[start][mid] < target) {
                    lo = mid + 1; // lo 变为mid的下一列/个
                } else if (matrix[start][mid] > target) {
                    hi = mid - 1;
                } else {
                    return true;
                }
            } else { // searching a row 查找行
            	// 纵向二分
                if (matrix[mid][start] < target) {
                    lo = mid + 1; // lo变为mid的下一行
                } else if (matrix[mid][start] > target) {
                    hi = mid - 1;
                } else {
                    return true;
                }
            }
        }

        return false;
    }

    public boolean searchMatrix(int[][] matrix, int target) {
        // an empty matrix obviously does not contain `target` 判断矩阵不为空
        if (matrix == null || matrix.length == 0) {
            return false;
        }

        // iterate over matrix diagonals 遍历次数是最小的一边。
        int shorterDim = Math.min(matrix.length, matrix[0].length);
        for (int i = 0; i < shorterDim; i++) {
            // 同时对行列进行二分查找，有一个true就返回true
            boolean verticalFound = binarySearch(matrix, target, i, true);
            boolean horizontalFound = binarySearch(matrix, target, i, false);
            if (verticalFound || horizontalFound) {
                return true;
            }
        }
        
        return false; 
    }
}

暴力法直接遍历所有元素也可以

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix[0].length; j++) {
                if (matrix[i][j] == target) {
                    return true;
                }
            }
        }

        return false;
    }
}