题目
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?
解答
这题和Pascal Triangle相近,要求返回第k行的数(注意:k从0开始,最开始被坑了T_T)。题目要求是空间复杂度为O(k),也就是说申请的数组或vector最多只能有k个元素。本着“先解决问题”的原则,第一次暂时不考虑这个条件,照搬Pascal Triangle的计算公式,利用递归(效率不高)的思路实现。思路很简单,直接上AC代码:
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> row;// the kth row
row.resize(rowIndex+1);
if (rowIndex == 0) {// the 1st row
row[0] = 1;
}
else if (rowIndex == 1) {// the 2nd row
row[0] = row[1] = 1;
}
else {// the other row
vector<int> rowPre;
rowPre.resize(rowIndex);
rowPre = getRow(rowIndex-1);
row[0] = row[row.size()-1] = 1;//take care of the begin and end
for(int i=1; i<row.size()-1; i++) {
row[i] = rowPre[i-1] + rowPre[i];
}
}
return row;
}
};
然后,再考虑空间复杂度的问题。自己没想出来,去评论区里看牛人们的解法,一开始也不知所云。只好采用带入具体情况去推理的笨办法,总算是把思路理清楚了,分析如下:
本质上,求出第k行的值需要依赖第(k-1)行的值。在上面的笨方法中,另外开辟了空间去保存第(k-1)行的值。其实,一旦确定要行号k,那么第k行的大小就确定为k+1。申请一个k+1大小的vector后,从第0行开始,一行一行地计算,前面行得值直接存储在最后要返回的k+1行的vector中。
要注意的是,由于如果沿袭”row[i]=row[i-1]+row[i]”的公式,从前往后去计算,那么在计算第i个元素值后,row[i]会被新值覆盖,但是计算第i+1个元素值时,又需要原来的row[i]。比如,要求第3行,现在求出了第2行的结果row[4]={1,2,1,0}。从前往后计算的话,row[4]={1,3,4,4};从后往前计算的话,row[4]={1,3,3,1}。因此,考虑从后往前计算,那么就不会存在覆盖的问题了。
分析略显啰嗦,主要是为了自己能看懂(^__^) 。第二次AC代码如下:
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> row;
row.resize(rowIndex+1);// the kth row has (k+1) elements
row[0] = 1;
for (int i=1; i<=rowIndex; i++) {// from 1st to kth row
for (int j=i; j>=1; j--) {// from row[k] to row[1]
row[j] += row[j-1];
}
}
return row;
}
};
小结:后面刷leetcode的原则还是“先解决,再优化”,目前coding水平不高,只有务实基础才是相对适合自己的方法。欢迎各位大牛拍砖指正,多多赐教!