51 Sort the Matrix Diagonally

2021-08-18 16:19:57 浏览数 (1)

题目

分析

题意:给一个m*n矩阵,按对角线进行升序排序。

将矩阵下标标出来

解题关键在于如何找到“对角线”。 我们人是可以直观看出来的,但是计算机不能。 通过观察发现,对角线的元素必有:i-j相同。 根据这个特性,实现对角遍历,将每个对角线视为一个数组

算法

代码语言:javascript复制
创建一个Map
遍历矩阵的每行每列
	如果在Map中不存在i-j这个key,就将其加入Map中
	将当前值加入到i-j这个数组中
对Map中所有的数组进行升序排序
遍历矩阵的每行每列,将值替换为Map中所有数组值

解答

使用优先队列(PriorityQueue)免去了排序过程,优先队列会自动排序;而且优先队列能够自动扩容,免去了动态求解对角线长度的困扰。

代码语言:javascript复制
class Solution {
    public int[][] diagonalSort(int[][] A) {
        int m = A.length, n = A[0].length;
        HashMap<Integer, PriorityQueue<Integer>> d = new HashMap<>();
        for (int i = 0; i < m;   i) {
            for (int j = 0; j < n;   j) {
                d.putIfAbsent(i - j, new PriorityQueue<>());
                d.get(i - j).add(A[i][j]);
            }
        }
        for (int i = 0; i < m;   i)
            for (int j = 0; j < n;   j)
                A[i][j] = d.get(i - j).poll();
        return A;
    }
    
}

0 人点赞