You are given an n x n 2D matrix representing an image.
Rotate the image by 90 degrees (clockwise).
Note:
You have to rotate the image in-place, which means you have to modify the input 2D matrix directly. DO NOT allocate another 2D matrix and do the rotation.
要求直接在此矩阵上进行修改
思路:从示例中我们可以看到,矩阵的第一行经过90度的翻转后变为最后一列,第二行变为倒数第二列,由此我们可以定义一个新的矩阵,根据新矩阵的内容来更新原矩阵的内容
#include<iostream>
#include<vector>
using namespace std;
void rotate(vector<vector<int>> &matrix){
vector<vector<int>> res;
res = matrix;
int n = matrix.size();
for(int i = 0;i<n;++i){
for(int j = 0;j<n;++j){
matrix[j][n-1-i] = data[i][j];
}
}
}
int main(){
int n;
while(cin>>n){
vector<vector<int>> data(n,vector<int>(n,0));//定义一个二维矩阵并初始化
for(int i =0;i<n;++i){
for(int j =0;j<n;++j)
cin>>data[i][j];
}
rotate(data);
for(int i = 0;i<n;++i){
for(int j = 0;j<n-1;++j){
cout<<data[i][j]<<" ";
}
cout<<data[i][n-1]<<endl;
}
}
return 0;
}
由以上的结果可知,在时间空间方面此方法都不是最优的。
还有另外一种方法是不用额外的空间,而是进行两次交换。
即首先转置,然后再对称交换
#include<iostream>
#include<vector>
using namespace std;
void rotate(vector<vector<int>> &matrix){
int n = matrix.size();
for(int i =0;i<n;++i)
for(int j =0;j<n;++j)
swap(matrix[i][j],matrix[j][i]);//进行转置
for(int i = 0; i<n;++i)
for(int j = 0;j<n/2;++j)
swap(matrix[i][j],matrix[i][n-1-j]);
}
int main(){
int n;
while(cin>>n){
vector<vector<int>> data(n,vector<int>(n,0));//定义一个二维矩阵并初始化
for(int i =0;i<n;++i){
for(int j =0;j<n;++j)
cin>>data[i][j];
}
rotate(data);
for(int i = 0;i<n;++i){
for(int j = 0;j<n-1;++j){
cout<<data[i][j]<<" ";
}
cout<<data[i][n-1]<<endl;
}
}
return 0;
}