使用 C++ 对矩阵的每个对角线元素按升序排序的程序
programmingpythonserver side programming
假设我们有 n x m 矩阵 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和矩阵mat
n := 行数和m := 列数
创建一个名为temp的数组
i:= si和j := sj，以及索引:= 0
当 i < n 且 j < m 时，执行
将 m[i, j] 插入到 temp 中，然后将 i 和 j 增加 1
对 temp 数组进行排序
设置 index := 0，i := si 且 j := sj
当 i < n 且 j < m
mat[i, j] := temp[index]
将 i、j 和 index 增加 1
从主方法中，执行以下操作 −
n := 行数和 m := 列数
对于范围从 0 到 n – 1 的 i，执行
solve(i, 0, mat)
对于范围从 1 到 m – 1，执行
solve(0，j，mat)
返回 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, ],]
