牛客题霸 [顺时针旋转矩阵] C++题解/答案

题目描述

有一个NxN整数矩阵,请编写一个算法,将矩阵顺时针旋转90度。

给定一个NxN的矩阵,和矩阵的阶数N,请返回旋转后的NxN矩阵,保证N小于等于300。

题解:

根据题目要求,先将同一列的第i行与第n-1-i行互换,再按照对角线互换

代码:

class Rotate {
   
public:
    vector<vector<int> > rotateMatrix(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=0;j<i;j++)
            {
   
                swap(mat[i][j],mat[j][i]);
            }
        return mat;
    }
};