1. 程式人生 > >LeetCode119:楊輝三角 II解析

LeetCode119:楊輝三角 II解析

給定一個非負索引 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