LeetCode:数组(五)

本组囊括数组相关题目,且只包括两道有关杨辉三角的问题。

118. Pascal's Triangle

题目描述:简单

首先要知道杨辉三角:每个数等于它左上方和右上方两数之和,那么如果能够知道一行杨辉三角,我们就可以根据每对相邻的值轻松地计算出它的下一行。类似这种迭代的思想,官方也把它归到动态规划的思路。

其次我们使用一个两位数组triangle来存储杨辉三角每一行。

 1 class Solution:
 2     def generate(self, numRows: int) -> List[List[int]]:
 3         # 思路为杨辉三角的生成方式,下一行的数等于上一行的左上右上之和,使用动态规划思路求解:
 4         triangle = [] # 矩阵(二维数组),以一维数组为元素
 5         for i in range(numRows):
 6             row = [None for _ in range(i+1)] # 每一行有i+1个数字,先占位,row为一维数组
 7             row[0], row[-1] = 1, 1
 8 
 9             for j in range(1, i):
10                 row[j] = triangle[i-1][j-1] + triangle[i-1][j] # 矩阵为triangle,数组为row,row是矩阵的元素。
11             triangle.append(row)
12 
13         return triangle

 

时间复杂度:O(numRows^2):虽然更新 triangle 中的每个值都是在常量时间内发生的,但它会被执行 O(numRows^2)次。想要了解原因,就需要考虑总共有多少次循环迭代。很明显外层循环需要运行numRowsnumRows 次,但在外层循环的每次迭代中,内层循环要运行 rowNumrowNum 次。因此,triangle 发生的更新总数为1 + 2 + 3 + .. + n = n(n+1)/2,因此时间复杂度:O(numRows^2)。
空间复杂度:O(numRows^2):因为我们需要存储我们在 triangle 中更新的每个数字,所以空间需求与时间复杂度相同。

 

119. Pascal's Triangle II

题目描述:简单

和上一题类似,只不过这一题只需要返回杨辉三角的第K行而不是前K行所有。所有笨方法即是同前一题一样生成前K行,再取出最后一行即可,这样空间复杂度较高。

根据题意根本无须保存前K-1行所有的信息,只需要保存上一行的信息即可求出当前行;又因为更新当前行索引j的时候,就把之前j的信息覆盖掉了,而更新 j + 1 的时候又需要之前j的信息,所以在更新前,我们需要一个变量把之前j的信息保存起来。

第二种优化思路是常见的在第一种优化下考虑,既然需要从前往后需要覆盖可能会增加空间复杂度,那么从后往前就能避开这一点。因为在当前行中更新完j的信息后,虽然把j之前的信息覆盖掉了。但是下一次我们更新的是j - 1,需要的是j - 1和j - 2 的信息,j信息覆盖就不会造成影响了。


解法三:当前行元素倒数第一个肯定是1,从倒数第二个开始往前更新,第j-1个元素等于之前行这个位置元素加上之前行这个位置-1的元素之和:行[i] = 行[i] + 行[i-1]

 1 class Solution:
 2     def getRow(self, rowIndex: int) -> List[int]:
 3         row = [1 for _ in range(rowIndex+1)] # 占位,因为后面要相加,所有都初始化为1
 4         for i in range(rowIndex+1):
 5             row[-1] = 1 # 每行末尾为1
 6             for j in range(i-1, 0, -1): # 从倒数第二行开始更新
 7                 row[j] = row[j] + row[j-1]
 8         return row
 9         # 总结,这里有两个常见优化思路,即只保留上一层结果计算当层,从前往后更新,但要考虑覆盖的问题;
10         # 第二个是从后往前更新,就不用考虑覆盖问题。

 

时间复杂度:O(k²),两个循环
空间复杂度:只需第K行的数字信息,O(k)

posted @ 2020-11-10 10:15  Jesse-jia  阅读(100)  评论(0编辑  收藏  举报