算法竞赛入门经典第十章学习笔记杨辉三角形（Pascal's triangle）

最新推荐文章于 2024-05-14 16:10:18 发布

原创最新推荐文章于 2024-05-14 16:10:18 发布 · 435 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#算法 #数据结构

算法竞赛入门经典专栏收录该内容

13 篇文章

订阅专栏

这篇博客介绍了杨辉三角形的性质与算法竞赛中的应用。通过公式Cnk=(n−k+1)/k×Cnk−1，可以快速计算出第n行的组合数，避免了O(n^2)的时间复杂度。同时，讨论了在数列求和过程中，最终结果对m取余与哪些初始元素无关的问题，涉及杨辉三角的第n-1行与高精度计算方法。

杨辉三角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)$ 。其实我们只需要第n行，这种方法从第1行到第n-1行也都计算出来了。

我们知道杨辉三角形的第n行第k个元素的值为组合数中的 $C_n^k$ 。第n行除了第一个数，第k-1个元素的值为 $C_n^{k-1}$ 。又有 $C_n^k=n!/(k!×(n-k)!)$ ， $C_n^{k-1}=n!/((k-1)!×(n-k+1)!)$ 。很容易得到如下关系：

$C_n^k=(n-k+1)/k×C_n^{k-1}$

利用这个公式可以直接从第n行第一个元素开始递推。

#include<iostream>
#include<vector>
using namespace std;

void pascal(int n, vector<int> &v)
{
 for(int i=1;i<=n;i++)
   v[i]=v[i-1]*(n-i+1)/i;
}

int main()
{
  int n;
  cin>>n;
  vector<int> ans(n + 1,1);
  pascal(n, ans);
  for(int i = 0; i <= n; i++)
    cout<<ans[i]<<", ";
  cout<<endl;
  return 0;
}

10-4 无关的元素
对于给定的n个数 $a_1, a_2, a_3...a_n$ ，依次求出相邻两数之和，将得到一个新数列。重复上述操作，最后结果将变成一个数。问这个数除以m的余数和哪些数无关？例如n=3,m=2时，第一次求和得到 $a_1+a_2, a_2+a_3$ 。再求和得到 $a_1+2a_2+n_3$ 。它除以2的余数和 $a_2$ 无关。 $1≤n≤105,2≤m≤1091\leq n\leq 10^5, 2\leq m \leq 10^9$ 。

当n=2时，和为 $a_1+a_2$ ;
当n=3时，和为 $a_1+a_2)+(a_2+a_3)=a_1+2a_2+a_3$ ;
当n=4时，和为 $a_1+3a_2+3a_3+a_4$ ;
当n=5时，和为 $a_1+4a_2+6a_3+4a_4+a_5$ ;
…

将所有项加到最后的多项式系数是杨辉三角形的第n-1行（如果给出5个数，最后系数是杨辉三角形第4行（从0开始）：1，4，6，4，1)。问题转变为杨辉三角第n-1行哪些数是m的倍数。可以通过前面公式 $C_n^k=(n-k+1)/k×C_n^{k-1}$ 将所有系数推算出来。但是由于n值比较大，推算出来的系数溢出，不能直接将每个系数对m取余计算，可以用唯一分解法或高精度法。