Pascal's Triangle II

最新推荐文章于 2022-02-27 22:46:37 发布

原创最新推荐文章于 2022-02-27 22:46:37 发布 · 318 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#C++

Old Blogs 专栏收录该内容

111 篇文章

订阅专栏

本文介绍了一个高效的算法来获取帕斯卡三角形的第K行，该算法仅使用O(K)的空间复杂度。通过从后向前计算的方式，避免了完整构建整个帕斯卡三角形所需的额外空间。

Pascal's Triangle II
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?

思路：使用O(K)空间的话，就需要从后往前加

vector 初始化 vector<int> vint(rowIndex,1);

#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
    vector<int> getRow(int rowIndex) {
	rowIndex++;    
 	vector<int> vint(rowIndex,1);
 	if(rowIndex<=2)
		return vint;
	for(int i=2;i<rowIndex;++i)
		for(int j=i-1;j>=1;--j){
			vint[j]+=vint[j-1];
		}	
	return vint;
    }
};


int main(){
	Solution s;
	vector<int> vint = s.getRow(0);
	for(int i=0;i<vint.size();++i)
		cout<<vint[i]<<" ";
	cout<<endl;	
	vint = s.getRow(1);
	for(int i=0;i<vint.size();++i)
		cout<<vint[i]<<" ";
	cout<<endl;	
	vint = s.getRow(2);
	for(int i=0;i<vint.size();++i)
		cout<<vint[i]<<" ";
	cout<<endl;	
	vint = s.getRow(3);
	for(int i=0;i<vint.size();++i)
		cout<<vint[i]<<" ";
	cout<<endl;	
	vint = s.getRow(4);
	for(int i=0;i<vint.size();++i)
		cout<<vint[i]<<" ";
	cout<<endl;	
	return 0;
}