資料結構
網路
關係資料庫管理系統 (RDBMS)
作業系統
Java
MS Excel
iOS
HTML
CSS
Android
Python
C 語言程式設計
C++
C#
MongoDB
MySQL
Javascript
PHP
物理學
化學
生物學
數學
英語
經濟學
心理學
社會學
服裝設計
法學C++ 對角線排序矩陣
假設我們有一個 N x M 的矩陣，我們需要將其對角線按照從左上到右下的順序遞增排序。例如，如果矩陣如下：
| 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 := 列數
對於 i 的範圍從 0 到 n – 1
solve(i, 0, mat)
對於 j 的範圍從 1 到 m – 1
solve(0, j, mat)
返回 mat
示例 (C++)
讓我們來看下面的實現來更好地理解：
#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]]
更新於：2020年4月29日
瀏覽量：297
廣告