排序 + 模拟,LeetCode 1329. 将矩阵按对角线排序

一、题目

1、题目描述

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

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

2、接口描述

python3
复制代码
python 复制代码
class Solution:
    def diagonalSort(self, mat: List[List[int]]) -> List[List[int]]:
cpp
复制代码
cpp 复制代码
class Solution {
public:
    vector<vector<int>> diagonalSort(vector<vector<int>>& mat) {

    }
};

3、原题链接

1329. 将矩阵按对角线排序


二、解题报告

1、思路分析

遍历每一条对角线,放入临时数组排序完再放回去即可

2、复杂度

时间复杂度: O(mn log min(m, n)) 空间复杂度:O(min(m, n))

3、代码详解

python3
复制代码
python 复制代码
class Solution:
    def diagonalSort(self, mat: List[List[int]]) -> List[List[int]]:
        m, n = len(mat), len(mat[0])
        for i in range(m - 1, -1, -1):
            x, y = i, 0
            a = []
            while x < m and y < n:
                a.append(mat[x][y])
                x += 1
                y += 1
            x, y = i, 0
            for t in sorted(a):
                mat[x][y] = t
                x += 1
                y += 1
        for i in range(1, n):
            x, y = 0, i
            a = []
            while x < m and y < n:
                a.append(mat[x][y])
                x += 1
                y += 1
            x, y = 0, i
            for t in sorted(a):
                mat[x][y] = t
                x += 1
                y += 1

        return mat
cpp
复制代码
cpp 复制代码
class Solution {
public:
    vector<vector<int>> diagonalSort(vector<vector<int>>& mat) {
        int m = mat.size(), n = mat[0].size();
        for(int i = m - 1; ~i; i --){
            vector<int> a;
            for(int x = i, y = 0; x < m && y < n; x ++, y ++ )
                a.emplace_back(mat[x][y]);
            sort(a.begin(), a.end());
            for(int x = i, y = 0, t = 0; x < m && y < n; x ++, y ++, t ++)
                mat[x][y] = a[t];
        }
        for(int i = 1; i < n; i ++){
            vector<int> a;
            for(int x = 0, y = i; x < m && y < n; x ++, y ++ )
                a.emplace_back(mat[x][y]);
            sort(a.begin(), a.end());
            for(int x = 0, y = i, t = 0; x < m && y < n; x ++, y ++, t ++)
                mat[x][y] = a[t];
        }
        return mat;
    }
};
相关推荐
重庆小透明16 分钟前
力扣刷题记录【1】146.LRU缓存
java·后端·学习·算法·leetcode·缓存
desssq35 分钟前
力扣:70. 爬楼梯
算法·leetcode·职场和发展
clock的时钟1 小时前
暑期数据结构第一天
数据结构·算法
小小小小王王王2 小时前
求猪肉价格最大值
数据结构·c++·算法
岁忧2 小时前
(LeetCode 面试经典 150 题 ) 58. 最后一个单词的长度 (字符串)
java·c++·算法·leetcode·面试·go
BIYing_Aurora2 小时前
【IPMV】图像处理与机器视觉:Lec13 Robust Estimation with RANSAC
图像处理·人工智能·算法·计算机视觉
martian6654 小时前
支持向量机(SVM)深度解析:从数学根基到工程实践
算法·机器学习·支持向量机
孟大本事要学习4 小时前
算法19天|回溯算法:理论基础、组合、组合总和Ⅲ、电话号码的字母组合
算法
??tobenewyorker4 小时前
力扣打卡第二十一天 中后遍历+中前遍历 构造二叉树
数据结构·c++·算法·leetcode
让我们一起加油好吗4 小时前
【基础算法】贪心 (二) :推公式
数据结构·数学·算法·贪心算法·洛谷