Leetcode篇:旋轉影象
阿新 • • 發佈:2018-11-16
@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