LeetCode 1329. 将矩阵按对角线排序（矩阵、排序）

最新推荐文章于 2024-06-06 11:58:07 发布

岁忧

最新推荐文章于 2024-06-06 11:58:07 发布

阅读量104

点赞数 3

分类专栏： LeetCode 文章标签： leetcode 矩阵算法 c++

本文链接：https://blog.csdn.net/weixin_46028214/article/details/138307875

版权

LeetCode 专栏收录该内容

266 篇文章 0 订阅

订阅专栏

1329. 将矩阵按对角线排序

在这里插入图片描述

思路：斜线方程y=x+a，变个形-a=x-y。只要在一条主对角线上的点,那么他们的坐标之差x-y就是一个固定值-a。利用这个就可以把主对角线上的元素放进同一个数组里v[i-j+m].push_back(mat[i][j]);然后我们再对每一个v[i]进行排序，然后放回即可。

class Solution {
public:
    vector<vector<int>> diagonalSort(vector<vector<int>>& mat) {
        int n=mat.size(),m=mat[0].size();
        vector<vector<int>> v(m+n+5);
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                v[i-j+m].push_back(mat[i][j]);
            }
        } 

        for(auto &t:v){
            sort(t.begin(),t.end());
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                mat[i][j]=v[i-j+m][0];
                v[i-j+m].erase(v[i-j+m].begin());//vector里删除某个迭代器指向的位置
            }
        }
        return mat;
    }
};

下面代码用的是降序排列，没用过，所以整上。

class Solution {
public:
    vector<vector<int>> diagonalSort(vector<vector<int>>& mat) {
        int n=mat.size(),m=mat[0].size();
        vector<vector<int>> v(m+n+5);
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                v[i-j+m].push_back(mat[i][j]);
            }
        } 

        for(auto &t:v){
            sort(t.rbegin(),t.rend());//降序排列
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                mat[i][j]=v[i-j+m].back();
                v[i-j+m].pop_back();
            }
        }
        return mat;
    }
};