---
# System prepended metadata

title: '1329. 将矩阵按对角线排序[矩陣][排序]'

---

# 1329. 将矩阵按对角线排序[矩陣][排序]
矩阵对角线 是一条从矩阵最上面行或者最左侧列中的某个元素开始的对角线，沿右下方向一直到矩阵末尾的元素。例如，矩阵 mat 有 6 行 3 列，从 mat[2][0] 开始的 矩阵对角线 将会经过 mat[2][0]、mat[3][1] 和 mat[4][2] 。

给你一个 m * n 的整数矩阵 mat ，请你将同一条 矩阵对角线 上的元素按升序排序后，返回排好序的矩阵。

 ![](https://i.imgur.com/3Vo7LSJ.png)


示例 1：



```
输入：mat = [[3,3,1,1],[2,2,1,2],[1,1,1,2]]
输出：[[1,1,1,1],[1,2,2,2],[1,2,3,3]]
```
示例 2：

```
输入：mat = [[11,25,66,1,69,7],[23,55,17,45,15,52],[75,31,36,44,58,8],[22,27,33,25,68,4],[84,28,14,11,5,50]]
输出：[[5,17,4,1,52,7],[11,11,25,45,8,69],[14,23,25,44,58,15],[22,27,31,36,50,66],[84,28,75,33,55,68]]
```

解題思路：

每一個對角線上的mat[i][j]都有相同的值i-j
對每一條對角線
將他們的陣列元素值取出放在一起，排序後再放回去陣列

```java=
class Solution {
    public int[][] diagonalSort(int[][] mat) {
        int m = mat.length, n = mat[0].length;
        HashMap<Integer, PriorityQueue<Integer>> map = new HashMap<>();
        for(int i=0;i<m;i++){
            for(int j =0;j<n;j++){
                if(!map.containsKey(i-j)){
                    map.put(i-j, new PriorityQueue<Integer>());
                
                }
                map.get(i-j).add(mat[i][j]);

            }
        }
        for(int i=0;i<m;i++){
            for(int j =0;j<n;j++){
                mat[i][j] = map.get(i-j).poll();
            }
        }
        return mat;
    }
}

```


