Rotate Image
You are given an n x n 2D matrix representing an image.
Rotate the image by 90 degrees (clockwise).
Follow up:
Could you do this in-place?
解题思路:
方法1:计算出原矩阵中的各个数值在新矩阵中的位置,然后一步到位转换为新的旋转矩阵
当n=1时,矩阵保持不变
当n=2时,原矩阵a与旋转矩阵b有以下关系:
b[0][0] = a[1][0]
b[1][0] = a[1][1]
b[1][1] = a[0][1]
b[0][1] = a[0][0]
在这里可以得到规律:b[i][j] = a[n-1-j][i]
当n>=3时,也满足以上规律。
在不考虑空间复杂度的情况下,只需要再构建一个二维数组b[n][n],利用公式b[i][j] = a[n-1-j][i],就可以得到旋转矩阵了。但是,题目要求在原
数组进行旋转。因此,可以先将焦点放在一个元素的旋转上,在不丢失数据的情况下,每个值的旋转涉及到4个数,并存在以下关系:
int temp = a[i][j];
当n=1时,不需旋转
当n=2时,只需完成a[0][0]的旋转
当n=3时,需要完成a[0][0]、a[0][1]的旋转
当n=5时,需要完成a[0][0至4],a[1][1至2]的旋转
得到规律:对于要旋转的数a[i][j]满足,i < n/2 且 i <= j < n-1-i
代码:
#include <iostream>
#include <vector>
using namespace std;
void rotate(vector< vector<int> >& matrix)
{
int n = matrix.size();
for(int i = 0; i < n/2; i++)
{
for(int j = i; j < n-1-i; j++)
{
int t = matrix[i][j];
matrix[i][j] = matrix[n-1-j][i];
matrix[n-1-j][i] = matrix[n-1-i][n-1-j];
matrix[n-1-i][n-1-j] = matrix[j][n-1-i];
matrix[j][n-1-i] = t;
}
}
}
int main()
{
vector< vector<int> > matrix;
vector<int> rng;
int n, m;
cin>>n;
for(int i = 0; i < n; i ++)
{
for(int j = 0; j < n; j ++)
{
cin>>m;
if(i == 0) rng.push_back(m);
else rng[j] = m;
}
matrix.push_back(rng);
}
rotate(matrix);
for(int i = 0; i < matrix.size(); i++)
{
for(int j = 0; j < matrix[i].size(); j++)
{
cout<<matrix[i][j]<<' ';
}
cout<<endl;
}
}
方法2:先得到原矩阵的转置矩阵,再从转置矩阵转到目标矩阵,如图: