题目:给你一幅由 N × N
矩阵表示的图像,其中每个像素的大小为 4 字节。请你设计一种算法,将图像旋转 90 度。
不占用额外内存空间能否做到?
思路:参考。将矩阵分为四块,我们考虑左上角那块区域的坐标转换。通过绘制图像可得在左上角区域的坐标[i,j],经旋转对应在其他区域的坐标为(顺时针)(j, n-i-1), (n-i-1,n-j-1), (n-j-1,i)。
那么现在捋一下如何原地操作元素:
枚举左上区域的所有位置,然后通过上面总结的位置关系直接交换元素。
对于一个位置 (i,j),需要交换三次:
swap(matrix[i][j], matrix[j][n-i-1]);
swap(matrix[i][j], matrix[n-i-1][n-j-1]);
swap(matrix[i][j], matrix[n-j-1][i]);
这里需要考虑左上角区域最大列的最大行的坐标范围。最大行:n/2。最大列:(n+1)/2。
C++实现如下。
class Solution {
public:
void rotate(vector<vector<int>>& matrix) {
int n=matrix.size();
for(int i=0;i<n/2;i++)
{
for(int j=0;j<(n+1)/2;j++)
{
swap(matrix[i][j],matrix[j][n-i-1]);
swap(matrix[i][j],matrix[n-i-1][n-j-1]);
swap(matrix[i][j],matrix[n-j-1][i]);
}
}
}
};