给你一幅由 N × N 矩阵表示的图像,其中每个像素的大小为 4 字节。请你设计一种算法,将图像旋转 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]
]
这道题是一道典型的关于matrix(矩阵)的题, 一些技巧也在其他关于矩阵的题中出现,比如怎么蛇形print出矩阵的内容(https://leetcode-cn.com/problems/spiral-matrix/)
这道题用到的一些小技巧:把一个matrix 看成 一堆层(layer), 给你一个n x n matrix, 要知道这个matrix 有几层 (怎么定义一个层 能让code 好写点)
在一层里怎么(swap)换数字
一些需要想的点(corner cases):如果最有一层里面只有一个element?
一个 n x n matrix 有 math.ceil(n / 2) 个 layers
从第0个layer(最外层)来表示
我来表示每层的方法:
四个值怎么换(rotate/swap)
from math import ceil
def rotateMatrix(matrix):
# dimension for a n by n matrix
n = len(matrix)
# e.g. if it has 2 layers we have layer number 0,1
# and the starting index for each layer is 0,0 1,1, 2,2....
for layer in range(0, ceil(n / 2)):
start = layer
# any element from start to element before end [start,end) forms a slice
# where we need to 4 way swap each elements
end = n - 1 - layer
i = start
while i < end:
tmpTop = matrix[layer][i]
matrix[layer][i] = matrix[n - 1 - i][layer]
matrix[n - 1 - i][layer] = matrix[n - 1 - layer][n - 1 - i]
matrix[n - 1 - layer][n - 1 - i] = matrix[i][n - 1 - layer]
matrix[i][n - 1 - layer] = tmpTop
i += 1
# constant space O(1), O(NxN)
# a less memory efficient way of doing this is to have an array of tmp values O(N) space
m = [[1,2,3],[4,5,6],[7,8,9]]
import copy
cm = copy.deepcopy(m)
rotateMatrix(m)
m2 = [[5,1,9,11],[2,4,8,10],[13,3,6,7],[15,14,12,16]]
cm2 = copy.deepcopy(m2)
rotateMatrix(m2)
问题来源:力扣(LeetCode)
链接:力扣
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。