Given an index k, return the kth row of the Pascal's triangle.
For example, given k = 3,
Return [1,3,3,1]
//方法一:利用两个vector,一个保存当前层的数,一个保存前面所有层的数
//空间复杂度较高
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> res;
vector<vector<int>> temp;
for (int i = 0; i <=rowIndex;++i)
{
res.clear();
if (i == 0) res.push_back(1);
else
{
for (int j = 0; j <= i; ++j)
{
if (j == 0 || j == i) res.push_back(1);
else
{
res.push_back(temp[i - 1][j - 1] + temp[i - 1][j]);
}
}
}
temp.push_back(res);
}
return res;
}
};
解题思路分析:
这道题跟 Pascal's Triangle 很类似,只是这里只需要求出某一行的结果。 Pascal's Triangle 中因为是求出全部结果,所以我们需要上一行的数据就很自然的可以去取。而这里我们只需要一行数据,就得考虑一下是不是能只用一行的空间来存储结果而不需要额外的来存储上一行呢?这里确实是可以实现的,对于每一行我们知道如果从前往后扫,第i个元素的值等于上一行的ans[i] + ans[i + 1],可以看到数据是往前看的,如果我们只用一行空间,那么需要的数据就会被覆盖掉。所以这里采取的方法是从后往前扫,这样每次需要的数据就是ans[i] + ans[i - 1],我们需要的数据不会被覆盖,因为需要的ans[i]只在当前步用,下一步就不需要了。这个技巧在动态规划省空间时也经常使用,主要就是看我们需要的数据是原来的数据还是新的数据来决定我们遍历的方向。时间复杂度还是O(n ^ 2),而空间这里是O(k)来存储结果,仍然不需要额外空间。
//方法二:从后往前计算
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> ans;
if (rowIndex < 0) return ans;
for (int i = 0; i <= rowIndex; ++i)
{
ans.push_back(0);
}
for (int j = 0; j <= rowIndex; ++j)
{
ans[rowIndex] = 1;
for (int k = rowIndex - 1; k > 0; --k) //从后往前
{
ans[k] = ans[k] + ans[k - 1];
}
ans[0] = 1;
}
return ans;
}
};