C++程式:對矩陣對角線元素進行升序排序
假設我們有一個n x m矩陣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,index := 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的j,執行以下操作:
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, ],]
廣告