【题目】
给定一个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,开始处理下一个子矩阵的外圈。原理同上。
下面是使用python3.5实现的代码。
def rotateMatrix(m):
tR = tC = 0
dR = len(m) - 1
dC = len(m[0]) - 1
while tR <= dR:
for i in range(tR, dR):
m[tR][tC+i], m[tR+i][dC], m[dR][dC-i], m[dR-i][tC] \
= m[dR-i][tC], m[tR][tC+i], m[tR+i][dC], m[dR][dC-i]
tR = tC = tR + 1
dR = dC = dR - 1