实习--Day17

8/13

今天可能还要继续看代码，但是也要对之前的代码进行复盘

先刷道题吧

旋转矩阵

链接：https://leetcode-cn.com/problems/rotate-matrix-lcci/

给你一幅由 N × N 矩阵表示的图像，其中每个像素的大小为 4 字节。请你设计一种算法，将图像旋转 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]
]

原地旋转矩阵：可以先进行对角元素调换，在对每一行进行reverse

    def rotate(matrix: List[List[int]]) -> None:
        """
        Do not return anything, modify matrix in-place instead.
        """
        for i in range(len(matrix)):
            for j in range(i):
                matrix[i][j],matrix[j][i] = matrix[j][i],matrix[i][j]
        for i in range(len(matrix)):
            matrix[i].reverse()
        return matrix

 

零矩阵

链接：https://leetcode-cn.com/problems/zero-matrix-lcci/

编写一种算法，若M × N矩阵中某个元素为0，则将其所在的行与列清零。

 

示例 1：

输入：
[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]
输出：
[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]
示例 2：

输入：
[
  [0,1,2,0],
  [3,4,5,2],
  [1,3,1,5]
]
输出：
[
  [0,0,0,0],
  [0,4,5,0],
  [0,3,1,0]
]

这题咋一看觉得很简单的一道题，但是实际写代码起来就会有各种问题，你可能会想到用map，但是map会生成的是一个拷贝，而不是原数组，如果用四层循环那一般超时了。

所以采用标记法是比较好的方案，就是记录为零的行号和列号，而且要用数组来存，因为可能有很多个零

    def setZeroes(self, matrix: List[List[int]]) -> None:
        """
        Do not return anything, modify matrix in-place instead.
        """
        # 记录为零的某行某列，做标记
        import functools
        if not matrix:
            return []
        row = [0]*(len(matrix))
        column=[0]*(len(matrix[0]))
        for i in range(len(matrix)):
            for j in range(len(matrix[0])):
                if matrix[i][j] == 0:
                    row[i] = 1
                    column[j] = 1
        for i in range(len(matrix)):
            for j in range(len(matrix[0])):
                if row[i]==1 or column[j]==1:
                    matrix[i][j] = 0
        
        return matrix