码迷,mamicode.com
首页 > 编程语言 > 详细

leetcode 【 Pascal's Triangle II 】python 实现

时间:2015-01-18 22:30:51      阅读:291      评论:0      收藏:0      [点我收藏+]

标签:

题目

Given an index k, return the kth row of the Pascal‘s triangle.

For example, given k = 3,
Return [1,3,3,1].

Note:
Could you optimize your algorithm to use only O(k) extra space?

 

代码:oj测试通过 Runtime: 48 ms

 1 class Solution:
 2     # @return a list of integers
 3     def getRow(self, rowIndex):
 4         if rowIndex == 0:
 5             return [1]
 6         if rowIndex == 1:
 7             return [1,1]
 8         pascal = [1,1]
 9         for i in range(1,rowIndex):
10             for j in range(len(pascal)-1):
11                 pascal[j] = pascal[j] + pascal[j+1]
12             pascal.insert(0,1)
13         return pascal

思路:

先把special case单列出来

每轮迭代从前向后更新数组的每个元素

最后再在第一位补上一个1

另,网上有很多题解是每轮从后往前遍历,这样的效率似乎更高一些。

后续又把遍历的顺序改为由后向前,结果如下

oj测试通过  Runtime: 37 ms

代码

 1 class Solution:
 2     # @return a list of integers
 3     def getRow(self, rowIndex):
 4         if rowIndex == 0:
 5             return [1]
 6         if rowIndex == 1:
 7             return [1,1]
 8         pascal = [1,1]
 9         
10         # from start to end
11         #for i in range(1,rowIndex):
12         #    for j in range(len(pascal)-1):
13         #        pascal[j] = pascal[j] + pascal[j+1]
14         #    pascal.insert(0,1)
15         
16         # from end to start
17         for i in range(1, rowIndex):
18             for j in range(len(pascal)-1, 0, -1):
19                 pascal[j] = pascal[j] + pascal[j-1]
20             pascal.insert(len(pascal),1)
21         return pascal

确实从后向前遍历比从前向后遍历要快很多。

小白对Python原理不懂,分析可能的原因如下:

1. 在数据起始位置insert,需要改变数组后续所有元素在内存中的位置,因此耗时长?

2. 在数组末尾insert,不需要改变数组之前所有元素在内存中的位置,只补上最后一个新增元素的位置就行了,因而速度快?

leetcode 【 Pascal's Triangle II 】python 实现

标签:

原文地址:http://www.cnblogs.com/xbf9xbf/p/4232335.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!