LeetCode119:楊輝三角 II解析
阿新 • • 發佈:2018-11-11
給定一個非負索引 k,其中 k ≤ 33,返回楊輝三角的第 k 行。
在楊輝三角中,每個數是它左上方和右上方的數的和。
示例:
輸入: 3
輸出: [1,3,3,1]
進階:
你可以優化你的演算法到 O(k) 空間複雜度嗎?
這個題和118題沒什麼太大的區別,只是希望可以優化一下空間複雜度。也就是說之前是要儲存整個三角,而現在只存之前的那一行,這個稍微改一點點就可以,其實就是動態規劃的思想。
C++原始碼:
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector< int> pre;
pre.push_back(1);
if (rowIndex==0) return pre;
pre.push_back(1);
if (rowIndex==1) return pre;
for (int i=2;i<=rowIndex;i++)
{
vector<int> now;
now.push_back(1);
for (int j=1;j<i;j++)
now. push_back(pre[j-1]+pre[j]);
now.push_back(1);
pre = now;
}
return pre;
}
};
python3原始碼:
class Solution:
def getRow(self, rowIndex):
"""
:type rowIndex: int
:rtype: List[int]
"""
l = [1]
if rowIndex== 0:
return l
l.append(1)
if rowIndex==1:
return l
for i in range(2, rowIndex+1):
temp = [1]
for j in range(1, i):
temp.append(l[j-1]+l[j])
temp.append(1)
l = temp
return l