1329.Sort the Matrix Diagonally

1329.Sort the Matrix Diagonally

难度:Medium

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

输入: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

思路: 分两种情况,一种是宽大于高,一种是宽小于高。 排序分三步,第一部分是左下部分的对角线排序,第二部分是右上部分的对角线排序,第三部分是中间部分对角线排序。

class Solution {
public:
    vector<vector<int>> diagonalSort(vector<vector<int>>& mat) {
        int m=mat.size();
        int n=mat[0].size();
 
        
        if(m>n)
        {
        for(int k=m-2;k>m-n;k--)
        {
            for(int i=k;i<m-1;i++)
            for(int j=i+1;j<m;j++)
            {
                if(mat[i][i-k]>mat[j][j-k])
                swap(mat[i][i-k], mat[j][j-k]);
            }   
        } 
        for(int k=n-2;k>0;k--)
        {    
            for(int i=k;i<n-1;i++)
            for(int j=i+1;j<n;j++)
            {
                if(mat[i-k][i]>mat[j-k][j])
                swap(mat[i-k][i], mat[j-k][j]);
            }        

        }

             for(int k=0;k<=m-n;k++)
            {
                for(int i=0;i<n-1;i++)
                for(int j=i+1;j<n;j++)
                if(mat[i+k][i]>mat[j+k][j])
                swap(mat[i+k][i], mat[j+k][j]);
            } 
        }
        else 
        {
        for(int k=m-2;k>0;k--)
        {
            for(int i=k;i<m-1;i++)
            for(int j=i+1;j<m;j++)
            {
                if(mat[i][i-k]>mat[j][j-k])
                swap(mat[i][i-k], mat[j][j-k]);
            }   
        } 
        for(int k=n-m+1;k<n-1;k++)
        {    
            for(int i=k;i<n-1;i++)
            for(int j=i+1;j<n;j++)
            {
                if(mat[i-k][i]>mat[j-k][j])
                swap(mat[i-k][i], mat[j-k][j]);
            }        

        }
            for(int k=0;k<=n-m;k++)
            {
                for(int i=0;i<m-1;i++)
                for(int j=i+1;j<m;j++)
                if(mat[i][i+k]>mat[j][j+k])
                swap(mat[i][i+k], mat[j][j+k]);
            } 
        }
        return mat;

    }
};

执行用时 :24 ms, 在所有 C++ 提交中击败了21.23%的用户 内存消耗 :8.4 MB, 在所有 C++ 提交中击败了100.00%的用户

Last updated