码迷,mamicode.com
首页 > 其他好文 > 详细

119. Pascal's Triangle II

时间:2017-06-12 22:32:13      阅读:209      评论:0      收藏:0      [点我收藏+]

标签:lin   var   span   solution   ems   oge   question   keep   ast   

https://leetcode.com/problems/pascals-triangle-ii/#/solutions

 

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?

 

 
Sol:
 
Elegent! 
 
Let i be the ith row. Alawys append 1 to the end of the row before moving on the the next row.  
 
For each line, the jth element equals to the sum of jth element in the previous row and the (j-1)th element in the previous row.
 
We do not need to create a tuple to keep track of (i th row, j th element). Before value of j th element in ith row -- res[j] -- is updated, variable res[j] stores the value of j th element in (i-1) th row. And the res[j-1] stores the value of (j-1) th element in (i-1) th row.
 
Thus, we simply add them together to get the j th element in i th row. 
 
  
class Solution(object):
    def getRow(self, rowIndex):
        """
        :type rowIndex: int
        :rtype: List[int]
        """
        
        # O(k) space, O(n^2) time
        
        res = []
        for i in range(rowIndex+1):
            res.append(1)
            
            #range(start,end,step)
            for j in range(i,0,-1):
                if j == i:
                    res[j] = 1
                else:
                    res[j] += res[j-1]
        return res

 

Note:

 

1 When the index j th element j equals to the row numer i, it means the j th  element is the last element in the row.  

119. Pascal's Triangle II

标签:lin   var   span   solution   ems   oge   question   keep   ast   

原文地址:http://www.cnblogs.com/prmlab/p/6995055.html

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