题目描述
给你一个大小为 m x n 的整数矩阵 grid ,其中 m 和 n 都是 偶数 ;另给你一个整数 k 。
矩阵由若干层组成,如下图所示,每种颜色代表一层:
矩阵的循环轮转是通过分别循环轮转矩阵中的每一层完成的。在对某一层进行一次循环旋转操作时,层中的每一个元素将会取代其 逆时针 方向的相邻元素。轮转示例如下:
返回执行 k 次循环轮转操作后的矩阵。
示例 1:
输入:grid = [[40,10],[30,20]], k = 1
输出:[[10,20],[40,30]]
解释:上图展示了矩阵在执行循环轮转操作时每一步的状态。
示例 2:
输入:grid = [[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]], k = 2
输出:[[3,4,8,12],[2,11,10,16],[1,7,6,15],[5,9,13,14]]
解释:上图展示了矩阵在执行循环轮转操作时每一步的状态。
提示:
m == grid.length
n == grid[i].length
2 <= m, n <= 50
m 和 n 都是 偶数
1 <= grid[i][j] <= 5000
1 <= k <= 109
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/cyclically-rotating-a-grid
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
class Solution {
public:
void help(vector<vector<int>>& grid, int x, int y, int m, int n){
int tmp = grid[x][y];
int i = x, j = y;
while(j < y+n-1){
grid[i][j] = grid[i][j+1];
j++;
}
while(i < x+m-1){
grid[i][j] = grid[i+1][j];
i++;
}
while(j > y){
grid[i][j] = grid[i][j-1];
j--;
}
while(i > x){
grid[i][j] = grid[i-1][j];
i--;
}
grid[i+1][j] = tmp;
}
vector<vector<int>> rotateGrid(vector<vector<int>>& grid, int k) {
int m = grid.size();
int n = grid[0].size();
int idx = 0;
while(m >= 2 && n >= 2){
int total = m*2 + (n-2)*2;
int round = k % total;
while(round > 0){
help(grid, idx, idx, m, n);
round--;
}
idx++;
m -= 2;
n -= 2;
}
return grid;
}
};