将正方形矩阵顺时针旋转90度

题目

给定一个矩阵matrix，把矩阵顺时针转动90度：
1 2 3
4 5 6
7 8 9
->
7 4 1
8 5 2
9 6 3

思路

比较简单的思路
1）先上下对称
->
7 8 9
4 5 6
1 2 3
2）对角互换
->
7 4 1
8 5 2
9 6 3

实现

def rotate(matrix):
    if matrix is None or len(matrix) == 0 or len(matrix[0]) != len(matrix):
        return

    size = len(matrix)
    for i in range(size//2):
        matrix[i], matrix[size-i-1] = matrix[size-i-1], matrix[i]

    for i in range(size):
        for j in range(i):
            matrix[i][j], matrix[j][i] = matrix[j][i], matrix[i][j]

    return matrix

测试

def test_rotate(size):
    n = 1
    mat = []
    for i in range(size):
        row = [n+j for j in range(size)]
        n += size
        mat.append(row)

    print(mat)
    rotate(mat)
    print(mat)


if __name__ == '__main__':
    test_rotate(3)

结果

[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
[[7, 4, 1], [8, 5, 2], [9, 6, 3]]

测试2

牛客网-将正方形矩阵顺时针旋转90度