leetcode 1329.矩阵按对角线排序

原题如下

https://leetcode-cn.com/problems/sort-the-matrix-diagonally/

题解

方法一

找出所有对角线元素，组成数组，排序后再放回去，遍历对角线的分参照 leetcode 766.托普利茨矩阵，另外此方法为原地修改矩阵，额外空间仅为长度为min（长，宽）的数组。
本思路java代码：

/*
 *作者@v7fgg
 *执行用时 :3 ms, 在所有 Java 提交中击败了85.29%的用户
 *内存消耗 :40.5 MB, 在所有 Java 提交中击败了100.00%的用户
 *2020年6月12日 14:47
 */
class Solution {
    public int[][] diagonalSort(int[][] mat) {
        int chang=mat.length;
        int kuan=mat[0].length;
        if(chang<kuan){
            for(int i=0;i<chang-1;i++){
                int[] m=new int[i+1];
                for(int j=0;j<=i;j++){
                    m[j]=mat[chang-1-i+j][j];
                }
                Arrays.sort(m);
                for(int j=0;j<=i;j++){
                    mat[chang-1-i+j][j]=m[j];
                }
            }
            for(int i=0;i<=kuan-chang;i++){
                int[] m=new int[chang];
                for(int j=0;j<chang;j++){
                    m[j]=mat[j][i+j];
                }
                Arrays.sort(m);
                for(int j=0;j<chang;j++){
                    mat[j][i+j]=m[j];
                }
            }
            for(int i=0;i<chang-1;i++){
                int m[]=new int[chang-1-i];
                for(int j=0;j<chang-1-i;j++){
                    m[j]=mat[j][kuan-chang+1+i+j];
                }
                Arrays.sort(m);
                for(int j=0;j<chang-1-i;j++){
                    mat[j][kuan-chang+1+i+j]=m[j];
                }
            }
        }
        else{
            for(int i=0;i<kuan-1;i++){
                int m[]=new int[i+1];
                for(int j=0;j<=i;j++){
                    m[j]=mat[chang-1-i+j][j];
                }
                Arrays.sort(m);
                for(int j=0;j<=i;j++){
                    mat[chang-1-i+j][j]=m[j];
                }
            }
            for(int i=0;i<=chang-kuan;i++){
                int m[]=new int[kuan];
                for(int j=0;j<kuan;j++){
                    m[j]=mat[chang-kuan-i+j][j];
                }
                Arrays.sort(m);
                for(int j=0;j<kuan;j++){
                    mat[chang-kuan-i+j][j]=m[j];
                }
            }
            for(int i=0;i<kuan-1;i++){
                int m[]=new int[kuan-1-i];
                for(int j=0;j<kuan-i-1;j++){
                    m[j]=mat[j][i+1+j];
                }
                Arrays.sort(m);
                for(int j=0;j<kuan-i-1;j++){
                    mat[j][i+1+j]=m[j];
                }
            }
        }return mat;
    }
}