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