排序 + 模拟,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;
    }
};
相关推荐
无限进步_34 分钟前
扫雷游戏的设计与实现:扫雷游戏3.0
c语言·开发语言·c++·后端·算法·游戏·游戏程序
qq_4335545438 分钟前
C++ 完全背包
开发语言·c++·算法
lingran__43 分钟前
算法沉淀第二天(Catching the Krug)
c++·算法
im_AMBER2 小时前
杂记 15
java·开发语言·算法
爱coding的橙子2 小时前
每日算法刷题Day70:10.13:leetcode 二叉树10道题,用时2h
算法·leetcode·深度优先
ghie90902 小时前
基于MATLAB的遗传算法优化支持向量机实现
算法·支持向量机·matlab
朝新_3 小时前
【优选算法】第一弹——双指针(上)
算法
艾莉丝努力练剑3 小时前
【C++STL :stack && queue (一) 】STL:stack与queue全解析|深入使用(附高频算法题详解)
linux·开发语言·数据结构·c++·算法
CoovallyAIHub3 小时前
ICLR 2026 惊现 SAM 3,匿名提交,实现“概念分割”,CV领域再迎颠覆性突破?
深度学习·算法·计算机视觉
IT古董4 小时前
【第五章:计算机视觉-计算机视觉在工业制造领域中的应用】1.工业缺陷分割-(2)BiseNet系列算法详解
算法·计算机视觉·制造