题目
分析
题意:给一个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;
}
}