leetcode 1329.将矩阵按对角线排序

题目描述:

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

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

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

解题思路:

一开始我想再建立一个相同的数组,然后把排序好的值填充到新数组里,这样不仅增大了空间复杂度,还增加了时间复杂度,因为要循环把对角线上的数提取出来,还要排序后再循环填充进去。
后来想到用冒泡排序,冒泡排序的关键是搞清楚要循环几次,这里我们发现,当行数大于列数时,就冒列数次,当列数大于行数时,要冒行数次,所以要循环min(行数,列数)次。一般的冒泡排序,是在每一次循环内,都要遍历一遍整个数组,使最小的数浮上来。这里因为要对多个对角线上的元素进行排序,所以每一条对角线都要看做一组待排序的数据,要遍历所有的对角线。
所以这样就很清晰了,外部大循环是min(行数,列数)次,内部是遍历所有的对角线,对于每一条对角线,都要从起始元素开始,和后面的元素比较、交换,使大的数沉下去,小的数浮上来。

c++代码:

class Solution {
public:
    vector> diagonalSort(vector>& mat) {
        for(int k=0;kmat[i+1][j+1]){
                        swap(mat[i][j],mat[i+1][j+1]);
                    }
                }
            }
        }
        return mat;
    }
};

你可能感兴趣的:(c++,数据结构,排序算法)