Hot100-Day07-T48旋转图像

0 阅读1分钟

Day07[26/3/7]T48.旋转图像

给定一个 n × n 的二维矩阵 matrix 表示一个图像。请你将图像顺时针旋转 90 度。

你必须在原地 旋转图像,这意味着你需要直接修改输入的二维矩阵。请不要 使用另一个矩阵来旋转图像。

示例 1:

输入:matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出:[[7,4,1],[8,5,2],[9,6,3]]

示例 2:

输入:matrix = [[5,1,9,11],[2,4,8,10],[13,3,6,7],[15,14,12,16]]
输出:[[15,13,2,5],[14,3,4,1],[12,6,8,9],[16,7,10,11]]

提示:

  • n == matrix.length == matrix[i].length
  • 1 <= n <= 20
  • -1000 <= matrix[i][j] <= 1000

解题思路

逆时针旋转 90° == 先水平翻转 + 再对角线翻转

类似的:

顺时针旋转 90° == 先竖直翻转 + 再对角线翻转

旋转 180° == 水平翻转 + 竖直翻转

Code

#include <iostream>
using namespace std;

#include <vector>
#include <algorithm>

class Solution
{
public:
    void rotate(vector<vector<int>> &matrix)
    {
        // 1. 竖直翻转
        for (int i = 0; i < matrix.size() / 2; i++)
        {
            swap(matrix[i], matrix[matrix.size() - 1 - i]);
        }

        // 2. 对角线翻转
        for (int i = 0; i < matrix.size(); i++)
        {
            for (int j = i + 1; j < matrix.size(); j++)
            {
                swap(matrix[i][j], matrix[j][i]);
            }
        }
    }
};

int main()
{

    // vector<vector<int>> matrix = {
    //     {1, 2, 3},
    //     {4, 5, 6},
    //     {7, 8, 9},
    // };

    vector<vector<int>> matrix = {
        {1, 2, 3},
        {4, 5, 6},
        {7, 8, 9}};

    Solution sol;

    sol.rotate(matrix);
    for (int i = 0; i < matrix.size(); i++)
    {
        cout << "[";
        for (int j = 0; j < matrix.size(); j++)
        {
            cout << matrix[i][j] << ",";
        }
        cout << "]" << endl;
    }
    return 0;
}