题目链接:https://leetcode.com/problems/pascals-triangle-ii/
Runtimes:3ms
1、问题
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?
2、分析
帕斯卡三角形,也叫杨辉三角形,如下图:
第0行开始,只有”1”,第1行有”1 1”,第2行有”1 2 1”,以此类推。
可见只要保留上一行,下一行即可求解。
3、小结
时间复杂度O(km),k表示所求行数,m表示需要一行需要计算的个数。空间复杂度O(k)。
4、实现
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector <int> v1, v2;
v1.push_back(1);
if(rowIndex == 0)
return v1;
v1.push_back(1);
for(int i = 2; i <= rowIndex; i++)
{
v2.push_back(1);
int colIndex = 0;
while(colIndex < v1.size() - 1)
{
v2.push_back(v1[colIndex] + v1[colIndex + 1]);
colIndex++;
}
v2.push_back(1);
v1 = v2;
v2.clear();
}
return v1;
}
};
5、反思
简单的一道题目,又用上了vector,要看stl了解vector如何实现。