LeetCode - 119. Pascal's Triangle II - 思路详解 - C++

题目

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?

翻译

假设有一个索引k，返回帕斯卡三角（杨辉三角）第k行。比如：
给定索引k。则返回【1,3,3,1】
注：
是否可以优化空间复杂度为O(k)

思路

我们来模拟下：
0，【1】
1，【1,1】
2，【1,2,1】
3，【1,3,3,1】
即第i行和第i-1行的关系为
a[i][j] = a[i-1][j-1] + a[i-1][j] 。开头和结尾处为1。不用计算。

代码

class Solution {
public:
    vector<int> getRow(int rowIndex) {
        vector<int> row;
        if(rowIndex < 0){
            return row;
        }

        for(int i = 1; i <= rowIndex + 1; ++i){
            vector<int> buf;

            for(int j = 0; j < i; j ++){
                if(i <= 2){
                    buf.push_back(1);
                }else{
                    if(j > 0 && j < i-1) {
                        int tmp = row[j-1] + row[j];
                        buf.push_back(tmp);
                    }else{
                        buf.push_back(1);
                    }
                }

            }

            row = buf;
        }
        return row;
    }
};