题目描述
有一副由NxN矩阵表示的图像,这里每个像素用一个int表示,请编写一个算法,在不占用额外内存空间的情况下(即不使用缓存矩阵),将图像顺时针旋转90度。
给定一个NxN的矩阵,和矩阵的阶数N,请返回旋转后的NxN矩阵,保证N小于等于500,图像元素小于等于256。
测试样例:
[[1,2,3],[4,5,6],[7,8,9]],3
返回:[[7,4,1],[8,5,2],[9,6,3]]
先从最简单的出发,使用额外空间,代码如下
vector<vector<int> > transformImage(vector<vector<int> > mat, int n) {
// write code here
vector<vector<int> > temp(n,vector<int>(n,0));
for(int i = 0; i<n ;++i)
{
for(int j = n-1;j>=0;--j)
{
temp[i][n-j-1] = mat[j][i];
}
}
return temp;
}
但题中明确不使用缓存矩阵,即要求使用值交换完成这个题,想一下矩阵顺时针旋转90度,实际上对应矩阵上的行变成列,对应于矩阵的四个数进行旋转,即左上->右上,右上->右下,右下->左下,左下->左上,每次我们只要保存左上临时变量即可完成一次旋转,代码如下
vector<vector<int> > transformImage(vector<vector<int> > mat, int n) {
// write code here
for (int i = 0; i < n/2; i++) //循环次数等于行数/2
{
for (int j = i; j < n - 1 - i; j++)
{
int k = mat[i][j];//保存左上
mat[i][j] = mat[n - 1 - j][i];//左上=左下
mat[n - 1 - j][i] = mat[n - 1 - i][n - 1 - j];//左下= 右下
mat[n - 1 - i][n - 1 - j] = mat[j][n - 1 - i];//右下= 右上
mat[j][n - 1 - i] = k;//右上=左上
}
}
return temp;
}
寻找规律,会发现,顺时针旋转90度相当于先上下翻转,再按照主对角线翻转
1 2 3 7 8 9 7 4 1
4 5 6 —> 4 5 6 —> 8 5 2
7 8 9 1 2 3 9 6 3
代码如下
vector<vector<int> > transformImage(vector<vector<int> > mat, int n) {
// write code here
for(int i = 0;i<n/2;++i)
{
for(int j = 0;j<n;++j)
{
swap(mat[i][j],mat[n-1-i][j]);
}
}
for(int i = 0;i<n;++i)
{
for(int j = i+1;j<n;++j)
{
swap(mat[i][j],mat[j][i]);
}
}
return mat;
}
如果是逆时针旋转90度,先对角线反转,再上下反转。