C++程式:對矩陣對角線元素進行升序排序


假設我們有一個n x m矩陣Mat,我們需要從左上角到右下角按遞增順序對該矩陣進行對角線排序,以便所有對角線上的元素都已排序。如果輸入矩陣如下:

3311
2212
1112

輸出矩陣將是:

1111
1222
1233

為了解決這個問題,我們將遵循以下步驟:

  • 定義一個名為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, ],]

更新於:2020年10月7日

451 次檢視

開啟你的職業生涯

完成課程獲得認證

開始學習
廣告