题目
思路
找规律~
- step1: 使用reverse()对整个二维数组进行翻转(使得末尾的行跑到前面,即由1,2,3; 4,5,6; 7,8,9变成7,8,9; 4,5,6; 1,2,3)
- step2: 两层for循环(i初始为0,j初始为i+1),交换,swap([i][j], [j][i])
时间复杂度为O(n^2),空间复杂度为O(1)
代码
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
void rotate(vector<vector<int>>& matrix) {
//=核心=======================
int n=matrix.size();
reverse(matrix.begin(),matrix.end());
for(int i=0;i<n;i++){
for(int j=i+1;j<n;j++){
swap(matrix[i][j],matrix[j][i]);
}
}
//==========================
// 用于测试输出的
// for(int i=0;i<n;i++){
// for(int j=0;j<n;j++){
// cout<<matrix[i][j]<<" ";
// }
// cout<<"; ";
// }
}
int main(){
int n;
cin>>n;//矩阵大小
vector<vector<int>> matrix(n);
for(int i=0;i<n;i++){
matrix[i].resize(n);
for(int j=0;j<n;j++){
cin>>matrix[i][j];
}
}
rotate(matrix);
return 0;
}