LeetCode（119） Pascal's Triangle II

题目

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?

分析

与上题LeetCode 118 Pascal’s Triangle本质相同；

需要注意的是，本题参数 k = 3 代表下标为3的那行元素，下标计算从0开始，也就是第4行；

AC代码

class Solution {
public:
vector<int> getRow(int rowIndex) {
if (rowIndex == 0)
return  vector<int>(1,1);

//存储上一行vector
vector<int> pre = getRow(rowIndex - 1);

//计算当前行
vector<int> cur(rowIndex + 1, 0);
//初始化首尾元素
cur[0] = 1;
cur[rowIndex] = 1;

//计算中间元素
for (int i = 1; i < rowIndex; ++i)
{
cur[i] = pre[i-1] + pre[i];
}//for

return cur;
}
};

GitHub测试程序源码