Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

โปรแกรมจัดเรียงองค์ประกอบแนวทแยงแต่ละองค์ประกอบในลำดับจากน้อยไปมากของเมทริกซ์ใน C++


สมมติว่าเรามีเมทริกซ์ 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, ],]