48 - 旋转图像 (Rotate Image)
问题描述:
给定一个 n × n 的二维矩阵,表示一个图像。将图像顺时针旋转 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]
]
思路:
找到每行每列旋转的规律,每次交换四个点,方阵旋转90度, A[0,0]
点旋转到A[0,n]
点,A[0,n]
点旋转到A[n,n]
点,A[n,n]
点旋转到A[n,0]
点,A[n,0]
点旋转到A[0,0]
点。
由此规律判断,当输入4*4数组时,每次旋转四个点,最外层旋转4次,即可将最外层旋转完毕,同理,内层四个元素需旋转一次。
代码:
class Solution:
def rotate(self, matrix):
for i in range(0,int(len(matrix)/2)):
start = i
end = len(matrix) - i -1
for j in range(0,end-start):
temp = matrix[start][start+j]
matrix[start][start+j] = matrix[end-j][start]
matrix[end-j][start] = matrix[end][end-j]
matrix[end][end-j] = matrix[start + j][end]
matrix[start+j][end] = temp
print(matrix)
if __name__ == '__main__':
s = Solution()
s.rotate([
[ 5, 1, 9,11],
[ 2, 4, 8,10],
[13, 3, 6, 7],
[15,14,12,16]
])