将矩阵按对角线排序

目录

题目描述

解法

---

链接：https://leetcode.cn/problems/sort-the-matrix-diagonally/

题目描述

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

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

解法

java

class Solution {
    public int[][] diagonalSort(int[][] mat) {
        int m = mat.length, n = mat[0].length;
        for(int k=0;k<Math.min(n,m)-1;k++){
            for(int i=0;i<m-1;i++){
                for(int j=0;j<n-1;j++){
                    if(mat[i][j]>mat[i+1][j+1]){
                        mat[i][j]=mat[i][j]^mat[i+1][j+1];
                        mat[i+1][j+1]=mat[i][j]^mat[i+1][j+1];
                        mat[i][j]=mat[i][j]^mat[i+1][j+1];
                    }
                }
            }
        }
        return mat;
    }
}

c++

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