1. 程式人生 > >Leetcode篇:旋轉影象

Leetcode篇:旋轉影象


@author: ZZQ
@software: PyCharm
@file: rotate.py
@time: 2018/11/16 15:41
要求:給定一個 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]]
思路:經過觀察以後發現,位於matrix[i][j]處的元素被移動到了matrix[j][row_len-i+1]處,所以先開闢新空間儲存原來的矩陣,然後再將所有的元素位置進行修改即可。


class Solution():
    def __init__(self):
        pass

    def rotate(self, matrix):
            """
            :type matrix: List[List[int]]
            :rtype: void Do not return anything, modify matrix in-place instead.
            """
            row_len = len(matrix[0])
            b = []
            for i in range(row_len):
                temp_b = []
                for j in range(row_len):
                    temp_b.append(matrix[i][j])
                b.append(temp_b)

            for i in range(row_len):
                for j in range(row_len):
                    matrix[j][row_len - i - 1] = b[i][j]
            return matrix