杨辉三角Pascal’s triangle)如下:
其中每个格子是上一行与它相邻两个格子的和,最左和最右的格子值为1。如何快速推算出杨辉三角的第n行排列(n从0开始)?
最容易想到的方法是由第一行开始递推,使用数组v储存结果,数组temp储存上一行:
void pascal(int n, vector<int> &v)
{
vector<int> temp(n+1, 1);
for(int i=1;i<=n;i++)
{
for(int j=1;j<i;j++)
v[j]=temp[j-1]+temp[j];
temp=v;
}
}
时间为 O ( n 2 ) O(n^2) O(n2)。其实我们只需要第n行,这种方法从第1行到第n-1行也都计算出来了。
我们知道杨辉三角形的第n行第k个元素的值为组合数中的 C n k C_n^k Cnk。第n行除了第一个数,第k-1个元素的值为 C n k − 1 C_n^{k-1} Cnk−1。又有 C n k = n ! / ( k ! × ( n − k ) ! ) C_n^k=n!/(k!×(n-k)!) Cnk=n!/(k!×(n−k)!), C n k − 1 = n ! / ( ( k − 1 ) ! × ( n − k + 1 ) ! ) C_n^{k-1}=n!/((k-1)!×(n-k+1)!) Cnk−1