jayati

Matrix Similarity After Cyclic Shifts

Nov 25th, 2023
720
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.47 KB | None | 0 0
  1. class Solution {
  2. public:
  3.     bool areSimilar(vector<vector<int>>& mat, int k) {
  4.     int m = mat.size();
  5.     int n = mat[0].size();
  6.  
  7.     std::vector<std::vector<int>> original = mat;
  8.  
  9.     for (int i = 0; i < m; ++i) {
  10.         if (i % 2 == 0) {
  11.             rotate(mat[i].begin(), mat[i].begin() + k % n, mat[i].end());
  12.         } else {
  13.             rotate(mat[i].rbegin(), mat[i].rbegin() + k % n, mat[i].rend());
  14.         }
  15.     }
  16.  
  17.     return mat == original;
  18. }
  19. };
Advertisement
Add Comment
Please, Sign In to add comment