力扣48旋转图像C++

题目

请点击

思路

找规律~

• 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;
}