给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 3
输出: [1,3,3,1]
进阶:
你可以优化你的算法到 O(k) 空间复杂度吗?
思路:
建立2重循环,第一个计算从1到rowIndex的每一层的杨辉三角数,第二个从i-1到0更新单个的数,因为如果从头更新的话,会破坏原有的数据,所以需要从尾部往前开始更新,更新完后,在末尾加入一个1
代码:
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> a;
for(int i=0;i<=rowIndex;i++) //第一个循环,从0到rowIndex更新没一层的杨辉三角数
{
for(int j=i-1;j>0;j--) //第二个循环,从尾到头更新元素
{
a[j]=a[j-1]+a[j];
}
a.push_back(1); //更新完后,在末尾接上1
}
return a;
}
};