题目描述为,给定一个数字k,返回第k行的杨辉三角形。
愚蠢的我,并不知道什么是杨辉三角形,大一的记忆已经丢失了,百度百科如下,
经过百度之后,了解了杨辉三角形,解题思路为:创建一个二维的向量,每个元素赋值为1,然后根据第i行的第J列等于第i-1行的第i列和第i-1列的和这个规则,对每个元素重新赋值,其余端点和结尾数都为1,这个无需改变,我的AC代码为:
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<vector<int> > triangle;
int i,j;
for(i=0;i<=rowIndex;i++)
{
vector<int> v;
for(j=0;j<=i;j++)
{
v.push_back(1);
}
triangle.push_back(v);
}
for(i=1;i<=rowIndex;i++)
{
for(j=1;j<i;j++)
{
triangle[i][j]=triangle[i-1][j-1]+triangle[i-1][j];
}
}
return triangle[rowIndex];
}
};
但是有个缺点是,我的空间复杂度为n^2,题目提示是否能优化为n,查看了讨论区之后,真的是感慨,大牛就是大牛呀,代码如下:
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> A(rowIndex+1, 0);
A[0] = 1;
for(int i=1; i<rowIndex+1; i++)
for(int j=i; j>=1; j--)
A[j] += A[j-1];
return A;
}
};
他的时间复杂度依旧是n^2,但是空间复杂度变为n,只用一个一维的向量来存储每一行,然后根据行数的变化,这个数组动态的改变,首先定义了端点为1,然后从后向前实现,这是因为要想实现A[j]+=A[j-1],需要先保证A[j-1]是不变的,所以从后向前,厉害呀。
补充:之前写的代码一直卡在了二维向量赋值上,导致程序无法运行,二维向量的正确方式如代码所示,我的错误是因为我按照s数组的方式赋值,没有保证triangle[i]是一个vector。