1329. Sort the Matrix Diagonally

问题：

对二维数组，对角线排序

Example 1:
Input: mat = [[3,3,1,1],[2,2,1,2],[1,1,1,2]]
Output: [[1,1,1,1],[1,2,2,2],[1,2,3,3]]
 

Constraints:
m == mat.length
n == mat[i].length
1 <= m, n <= 100
1 <= mat[i][j] <= 100

 解法：

分别拿出对角线元素，组成数列，进行sort排序，再存入数组。

同一对角线上的元素mat[i][j]，i-j是相同的。因此可以使用map的key来标记。value存同一对角线的元素。

方法1：c++的优先队列 priority_queue<int, vector<int>, greater<int>>

//默认是大顶堆 {5,4,3,2,1}
priority_queue<int> a; 
priority_queue<int, vector<int>, less<int> > a;

//小顶堆 {1,2,3,4,5}
priority_queue<int, vector<int>, greater<int> > c;

在使用 push(val)的时候，就已经开始一边插入一边排序了。

由于是队列，pop也将是从头到尾，由小到大弹出。

代码参考：

class Solution {
public:
    vector<vector<int>> diagonalSort(vector<vector<int>>& mat) {
        unordered_map<int, priority_queue<int, vector<int>, greater<int>>> d;
        int m=mat.size();
        int n=mat[0].size();
        for(int i=0; i<m; i++){
            for(int j=0; j<n; j++){
                d[i-j].push(mat[i][j]);
            }
        }
        for(int i=0; i<m; i++){
            for(int j=0; j<n; j++){
                mat[i][j]=d[i-j].top();
                d[i-j].pop();
            }
        }
        return mat;
    }
};

方法2：直接使用vector，push_back依次存入元素后，

使用sort排序，再使用pop_back从后往前弹出元素，由大到小。

代码参考：

class Solution {
public:
    vector<vector<int>> diagonalSort(vector<vector<int>>& mat) {
        unordered_map<int, vector<int>> d;
        unordered_map<int, vector<int>>::iterator it;
        int m=mat.size();
        int n=mat[0].size();
        for(int i=0; i<m; i++){
            for(int j=0; j<n; j++){
                d[i-j].push_back(mat[i][j]);
            }
        }
        for(it=d.begin(); it!=d.end(); it++){
            sort(it->second.begin(),it->second.end());
        }
        for(int i=m-1; i>=0; i--){
            for(int j=n-1; j>=0; j--){
                mat[i][j]=d[i-j].back();
                d[i-j].pop_back();
            }
        }
        return mat;
    }
};