forked from partho-maple/coding-interview-gym
-
Notifications
You must be signed in to change notification settings - Fork 80
/
Copy path118_Pascal's_Triangle.py
30 lines (26 loc) · 977 Bytes
/
118_Pascal's_Triangle.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
class Solution(object):
def generate(self, numRows):
"""
:type numRows: int
:rtype: List[List[int]]
"""
result = [[0 for x in range(y + 1)] for y in range(numRows)]
memo = {}
for row in range(numRows):
for column in range(row + 1):
num = self.generageNumber(row, column, memo)
result[row][column] = num
memo[str(row) + "-" + str(column)] = num
return result
def generageNumber(self, row, column, memo):
if column == 0 or row == column:
memo[str(row) + "-" + str(column)] = 1
return 1
key = str(row) + "-" + str(column)
if key in memo.keys():
return memo[str(row) + "-" + str(column)]
else:
return self.generageNumber(row - 1, column - 1, memo) + self.generageNumber(row - 1, column, memo)
sol = Solution()
triangle = sol.generate(5)
print("Result: ", triangle)