旋转矩阵

题目

面试题 01.07. 旋转矩阵

给你一幅由 N × N 矩阵表示的图像,其中每个像素的大小为 4 字节。请你设计一种算法,将图像旋转 90 度。

不占用额外内存空间能否做到?

示例 1:

给定 matrix = 
[
  [1,2,3],
  [4,5,6],
  [7,8,9]
],

原地旋转输入矩阵,使其变为:
[
  [7,4,1],
  [8,5,2],
  [9,6,3]
]

解法

原地旋转,操作矩阵。大概思路:一圈一圈旋转,每一圈里面再单个旋转;

代码如下:

class Solution {
public:
    void rotate(vector<vector<int>>& matrix) {
        int N = matrix.size();
        int up = 0;
        int down = N-1;
        int temp = 0;
        while(up<down){
            for(int i=up;i<down;++i){
                temp = matrix[up][i];
                matrix[up][i] = matrix[down-i+up][up];
                matrix[down+up-i][up] = matrix[down][down+up-i];
                matrix[down][down+up-i] = matrix[i][down];
                matrix[i][down] = temp;
            }
            up++;
            down--;
        }
    }
};