สมมติว่าเรามีเมทริกซ์ Mat ขนาด 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 และ matrix mat
-
n :=จำนวนแถว และ m :=จำนวนคอลัมน์
-
สร้างอาร์เรย์ที่เรียกว่า temp
-
i:=si และ j :=sj และดัชนี :=0
-
ในขณะที่ฉัน
-
ใส่ m[i, j] ลงใน temp แล้วเพิ่ม i และ j ขึ้น 1
-
-
จัดเรียงอาร์เรย์ชั่วคราว
-
ตั้งค่าดัชนี :=0, ผม :=si และ j :=sj
-
ในขณะที่ฉัน
-
mat[i, j] :=temp[ดัชนี]
-
เพิ่ม i, j และดัชนีขึ้น 1
-
-
จากวิธีหลัก ให้ทำดังนี้ −
-
n :=จำนวนแถว และ m :=จำนวนคอลัมน์
-
สำหรับฉันอยู่ในช่วง 0 ถึง n – 1 ทำ
-
แก้ (i, 0, mat)
-
-
สำหรับ j ในช่วง 1 ถึง m – 1 ทำ
-
แก้(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, ],]