题目:
给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 3
输出: [1,3,3,1]
分析:
杨辉三角下一层元素(除去第一个和最后一个)是上一层相应索引 i 和 i-1两个元素的和;
但是一层循环无法即控制层数,又控制当前层数的元素;
因此使用两层循环,分别对应层数 和 当前层的元素;
使用两个容器分别存储当前层的元素,和上一层的元素
见代码1
优化
能否使用一个容器来实现呢?
temp容器保存着上一层的元素,对于current容器,需要从前向后访问其每一个元素,并将对应值赋值到current中,如果反向遍历就不会发生对应位置current元素尚未被访问就被覆盖;
空间复杂度减半见代码2;
代码1:ac
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> temp(rowIndex+1, 1);
vector<int> current(rowIndex+1, 1);
for(int i = 1; i < rowIndex+1; i++)
{
for(int j = 1;j < i; j++)
{
current[j] = temp[j] + temp[j-1];
}
temp = current;
}
return current;
}
};
代码2:ac
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> ret(rowIndex+1, 1);
// i = 2代表第0层和第1层并不需要我们判断,直接填1就ok
for(int i = 2; i < rowIndex+1; i++)
{
// j = i-1:每一层中的最后一个元素也不需要计算;直接就是1
for(int j = i -1; j > 0; j--)
{
ret[j] = ret[j] + ret[j-1];
}
}
return ret;
}
};