欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

数组与字符串——旋转矩阵

程序员文章站 2022-03-21 17:26:07
...

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

不占用额外内存空间能否做到?
数组与字符串——旋转矩阵
解题思路

  • 在使用额外空间的情况下,可以建立一个与当前矩阵大小相同的矩阵。通过旋转矩阵得出当前元素[i,j]旋转后的位置为[j,n-1-i]。然后把得到的矩阵赋值给matrix。
  • 在不使用额外空间的情况下,可通过矩阵先上下翻转而后沿对角线翻转即可得到旋转矩阵,在翻转时注意i,j的范围。

代码

class Solution {
public:
    void rotate(vector<vector<int>>& matrix) {
        int n = matrix.size();
        int i, j;  //i控制交换的轮数,j控制每轮交换的个数
        
        //沿中心水平轴翻转一次
        for(i = 0; i < n/2; i++) {
            for(j = 0; j < n; j++) {
                swap(matrix[i][j], matrix[n-1-i][j]);
            }
        }
        //沿对角线翻转一次(即转置矩阵)
        for(i = 0; i < n; i++) {
            for(j = 0; j < i; j++) {
                swap(matrix[i][j], matrix[j][i]);
            }
        }
    }
};