C++で各対角要素を行列の昇順でソートするプログラム

n x mの行列Matがあるとすると、このMatを左上から右下に向かって昇順で斜めに並べ替えて、対角線のすべての要素を並べ替える必要があります。したがって、入力行列が-

出力行列は-

これを解決するには、次の手順に従います-

• Solve（）というメソッドを定義します。これには、si、sj、および行列マットが必要です

  。

• n：=行数およびm：=列数

• tempと呼ばれる配列を作成します

• i：=siおよびj：=sj、およびインデックス：=0

• i

  • m [i、j]をtempに挿入し、iとjを1増やします

• 一時配列を並べ替える

• set index：=0、i：=si and j：=sj

• i

  • mat [i、j]：=temp [index]

  • i、j、インデックスを1増やします

• mainメソッドから、次のようにします-

• n：=行数およびm：=列数

• 0からn– 1の範囲のiの場合、実行

  • 解決（i、0、マット）

• 1からm– 1の範囲のjの場合、実行

  • ソルブ（0、j、マット）

• リターンマット

理解を深めるために、次の実装を見てみましょう-

例

#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, ],]