写出杨辉三角的某一行 Pascal's Triangle II

题目源自于leetcode。

题目:Given an index k, return the kth row of the Pascal's triangle.For example, given k = 3, Return [1,3,3,1].

思路:

杨辉三角的每一行其实是二项式系数的序列。

二项式系数C(n,r),可以用排列组合里的数学公式直接求,但是对于计算机来讲乘法和除法速度慢,而且容易溢出。所以还是用递推公式求解,以空间换时间。

根据杨辉三角或者组合公式可以知道,递推公式为C(n,r)=C(n-1,r-1)+C(n-1,r)

对于给定的k,对应的行中有k+1个数。借助递推公式来求每个C(n,i)。

注意如果同时计算k行的所有值,用k个存储空间不够,因为递推公式需要上一轮的两个值,直接在原地求的话C(n-1,r-1)的值会先被覆盖掉。


代码:

class Solution {
public:
    vector<int> getRow(int rowIndex) {
		vector<int> result;
        if(rowIndex < 0)
            return result;

		int i,j,r;
		vector<int> v;
        for(r=0; r<=rowIndex; r++)
        {
                //计算组合数C(rowIndex, r)
				v.clear();
			    for (i = 0; i <= r; i++)  
					v.push_back(1);

				for (i = 1; i <= rowIndex - r; i++)  
					for (j = 1; j <= r; j++)  
						v[j] += v[j - 1];

			    result.push_back(v[r]);
        }
        return result;
    }
};


你可能感兴趣的:(写出杨辉三角的某一行 Pascal's Triangle II)