【题目】
给定一个N × N的矩阵 matrix,把这个矩阵调整成顺时针转动90°后的形式。
例如:
1 2 3 4
5 6 7 8
9 10 11 12
13 14 15 16
顺时针转动90°的结果为:
13 9 5 1
14 10 6 2
15 11 7 3
16 12 8 4
要求额外空间复杂度O(1)。
【基本思路】
分圈处理。在矩阵中用左上角(tR, tC)和右下角(dR, dC)就可以表示一个子矩阵。比如,题目中的矩阵,当(tR, tC) = (0,0), (dR, dC) = (3,3)。表示的子矩阵就是整个矩阵,它的最外圈的部分如下:
1 2 3 4
5 8
9 12
13 14 15 16
在这个外圈中,1,4,16,13为一组,然后让1占据4的位置,4占据16的位置,16占据13的位置,13占据1的位置,这样一组就调整完了。然后2,8,15,9为一组,继续调整。最后3,12,14,5为一组,继续调整。都调整完之后,这一圈就旋转完毕。令tR, tC加1,dR,dC减1,开始处理下一个子矩阵的外圈。原理同上。
#还是 先大圈后小圈的思路
#抓住对角点的规律
def RotateMatrix(arr):
tr = tc = 0
dr = len(arr)-1
dc = len(arr[0])-1
while tr<dr:
#因为等于的时候就一个数所以不用旋转了
for i in range(dc-tc):
tmp = arr[tr][tc+i]
arr[tr][tc+i] = arr[dr-i][tc]
arr[dr-i][tc] = arr[dr][dc-i]
arr[dr][dc-i] = arr[tr+i][dc]
arr[tr+i][dc] = tmp
tr= tc =tr+1
dc =dr = dc-1
return arr
if __name__ == '__main__':
li = [[1,2,3],
[4,5,6],
[7,8,9]]
a = RotateMatrix(li)
print(a)