问题描述:
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?
帕斯卡三角即是杨辉三角。这个问题是给出k值,然后计算第(k+1)行的各个数,以vector 形式返回。
第一行 1
第二行 1 1
第三行 1 2 1
第四行 1 3 3 1
….
题目中要求only O(k) extra space,即对空间复杂度有了要求。这就是说我们要重复利用相同的空间,直到计算出所需要的行。
苦思冥想没有想到好的解决办法,遂查看资料,看看前辈们怎么做的。
从众多前辈们的解题思路来看,大都是用了一种从后向前扫,逐渐覆盖上一行的数据同时计算下一行的数据存储的做法来做的。根据这个思路,写下代码如下:
vector<int> getRow(int rowIndex) {
vector<int> a(rowIndex + 1);
a[0] = 1;
for(int i = 1; i <= rowIndex; i++)
for(int j = i; j >= 0; j--)
if (j == i)//第i行,最右边的数据为1
a[j] = 1;
else if (j == 0)
a[j] = 1;//第i行,最左边的数据为1
else
a[j] = a[j-1] + a[j]; //第i行,第j个元素,等于i-1行中,第j个元素和第j-1个元素之和。然后覆盖掉i-1行中第j个元素。
return a;
}