假设我们有nxm矩阵Mat,我们必须从左上角到右下角按对角顺序对Mat进行递增排序,以便对角线上的所有元素都进行排序。所以如果输入矩阵像-
3 | 3 | 1 | 1 |
2 | 2 | 1 | 2 |
1 | 1 | 1 | 2 |
输出矩阵将是-
1 | 1 | 1 | 1 |
1 | 2 | 2 | 2 |
1 | 2 | 3 | 3 |
为了解决这个问题,我们将遵循以下步骤-
定义一个名为的方法solve()
,它将采用si,sj和矩阵矩阵
n:=行数,m:=列数
制作一个名为temp的数组
i:= si和j:= sj,索引:= 0
当i <n和j <m时
将m [i,j]插入到temp,然后将i和j加1
排序临时数组
设置索引:= 0,i:= si和j:= sj
当i <n和j <m
mat [i,j]:= temp [index]
将i,j和index增加1
从主要方法中,执行以下操作-
n:=行数,m:=列数
如果我的范围是0到n – 1,
解决(i,0,mat)
对于1到m – 1范围内的j,执行
解决(0,j,mat)
返回垫
让我们看下面的实现以更好地理解-
#include <bits/stdc++.h> using namespace std; void print_vector(vector<vector<auto> > v){ cout << "["; for(int i = 0; i<v.size(); i++){ cout << "["; for(int j = 0; j <v[i].size(); j++){ cout << v[i][j] << ", "; } cout << "],"; } cout << "]"<<endl; } class Solution { public: void solve(int si, int sj, vector < vector <int> > &mat){ int n = mat.size(); int m = mat[0].size(); vector <int> temp; int i = si; int j = sj; int idx = 0; while(i < n && j < m){ temp.push_back(mat[i][j]); i++; j++; } sort(temp.begin(), temp.end()); idx = 0; i = si; j = sj; while(i < n && j < m){ mat[i][j] = temp[idx]; i++; j++; idx++; } } vector<vector<int>> diagonalSort(vector<vector<int>>& mat) { int n = mat.size(); int m = mat[0].size(); for(int i = 0; i <n; i++){ solve(i, 0, mat); } for(int j = 1; j < m; j++){ solve(0, j, mat); } return mat; } }; main(){ vector<vector<int>> v = {{3,3,1,1},{2,2,1,2},{1,1,1,2}}; Solution ob; print_vector(ob.diagonalSort(v)); }
{{3,3,1,1},{2,2,1,2},{1,1,1,2}}
输出结果
[[1, 1, 1, 1, ], [1, 2, 2, 2, ], [1, 2, 3, 3, ],]