Leetcode 1329：将矩阵按对角线排序

题目描述

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

示例 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]]
 

提示：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sort-the-matrix-diagonally
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

class Solution {
public:
    vector<vector<int>> diagonalSort(vector<vector<int>>& mat) {
        if(mat.size()==0 || mat[0].size()==0) return mat;
        int m=mat.size(),n=mat[0].size();
        priority_queue<int,vector<int>,greater<int>> que;
        for(int i=0;i<m;++i){
            int ti=i,tj=0;
            while(ti<m && tj<n){
                que.push(mat[ti][tj]);
                ti++;
                tj++;
            }
            ti=i;tj=0;
            while(!que.empty()){
                mat[ti++][tj++]=que.top();
                que.pop();
            }
        }
        for(int j=1;j<n;++j){
            int ti=0,tj=j;
            while(ti<m && tj<n){
                que.push(mat[ti][tj]);
                ti++;
                tj++;
            }
            ti=0;tj=j;
            while(!que.empty()){
                mat[ti++][tj++]=que.top();
                que.pop();
            }
        }
        return mat;
    }
};