程序在C ++中以矩阵的升序对每个对角元素进行排序

假设我们有nxm矩阵Mat,我们必须从左上角到右下角按对角顺序对Mat进行递增排序,以便对角线上的所有元素都进行排序。所以如果输入矩阵像-

3311
2212
1112

输出矩阵将是-

1111
1222
1233

为了解决这个问题,我们将遵循以下步骤-

  • 定义一个名为的方法solve(),它将采用si,sj和矩阵矩阵

  • n:=行数,m:=列数

  • 制作一个名为temp的数组

  • i:= si和j:= sj,索引:= 0

  • 当i <n和j <m时

    • 将m [i,j]插入到temp,然后将i和j加1

  • 排序临时数组

  • 设置索引:= 0,i:= si和j:= sj

  • 当i <n和j <m

    • mat [i,j]:= temp [index]

    • 将i,j和index增加1

  • 从主要方法中,执行以下操作-

  • n:=行数,m:=列数

  • 如果我的范围是0到n – 1,

    • 解决(i,0,mat)

  • 对于1到m – 1范围内的j,执行

    • 解决(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, ],]
猜你喜欢