数组与字符串——旋转矩阵
程序员文章站
2022-03-21 17:26:07
...
没时间了,先上题!
给你一幅由 N × N 矩阵表示的图像,其中每个像素的大小为 4 字节。请你设计一种算法,将图像旋转 90 度。
不占用额外内存空间能否做到?
解题思路
- 在使用额外空间的情况下,可以建立一个与当前矩阵大小相同的矩阵。通过旋转矩阵得出当前元素[i,j]旋转后的位置为[j,n-1-i]。然后把得到的矩阵赋值给matrix。
- 在不使用额外空间的情况下,可通过矩阵先上下翻转而后沿对角线翻转即可得到旋转矩阵,在翻转时注意i,j的范围。
代码
class Solution {
public:
void rotate(vector<vector<int>>& matrix) {
int n = matrix.size();
int i, j; //i控制交换的轮数,j控制每轮交换的个数
//沿中心水平轴翻转一次
for(i = 0; i < n/2; i++) {
for(j = 0; j < n; j++) {
swap(matrix[i][j], matrix[n-1-i][j]);
}
}
//沿对角线翻转一次(即转置矩阵)
for(i = 0; i < n; i++) {
for(j = 0; j < i; j++) {
swap(matrix[i][j], matrix[j][i]);
}
}
}
};