51 Sort the Matrix Diagonally

题目

分析

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

将矩阵下标标出来

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

算法

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

解答

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

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;
    }
    
}