Time: 20 ms (96.30%), Space: 14 MB (40.46%) - LeetHub

gkapelyushok · Apr 11, 2022 · 5626742 · 5626742
1 parent 894789c
commit 5626742
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/1260-shift-2d-grid/1260-shift-2d-grid.cpp b/1260-shift-2d-grid/1260-shift-2d-grid.cpp
@@ -0,0 +1,12 @@
+class Solution {
+public:
+    vector<vector<int>> shiftGrid(vector<vector<int>>& grid, int k) {
+        int n = grid.size();
+        int m = grid[0].size();
+        vector<vector<int>> result(n, vector<int>(m));
+        for (int i = 0; i < n * m; i++) {
+           result[(i + k) % (n * m) / m][(i + k) % (n * m) % m] = grid[i / m][i % m];
+        }
+        return result;
+    }
+};