题目来自PTA。
二叉搜索树定义为:
左子树仅包含小于根结点的元素;
右子树仅包含大于根结点的元素;
左右子树均为二叉搜索树。
一棵二叉搜索树可以有不同的插入顺序。例如,对于如下的二叉搜索树
插入序列可以是 3 2 1 4 6 5, 也可以是 3 2 4 1 6 5 但不可以是 3 2 4 5 6 1。
下面给出一棵二叉搜索树的先序遍历序列,请编写程序求出该树的插入序列个数。考虑到总数可能非常大,请输出总数对 1000000007 (10^9 +7)取余的结果。
输入格式:
第一行给出一个整数 N (0<N<=100),表示二叉树中元素个数;
第二行给出 N 个正整数,以空格分隔,表示二叉树的先序遍历序列;
输出格式:
在一行中输出插入序列总数。
输入样例:
6
3 2 1 4 6 5
输出样例:
10
分析
其实这道题的关键在于求出所有可能的排序,实际上是个组合问题。如果想知道详细求出公式的过程可以自行搜索插空法。
假设一个树的左子树节点数n,左子树符合条件的排列种数为a;其右子树节点数为m,符合条件的排列种数为b。那么该树符合要求的排列总数为
a * b * C(m + n,m);
C(m + n,m)表示从m+n中取m个的组合数。
每一个左子树的排列数可以通过它自己的左右子树用同样的公式获得。也就是说,只需要使用递归便可以从底下往上依次求得每个节点的排列数,最后汇总到根节点。
之后问题就变为了求组合数的问题,如果使用组合数的公式,很容易就溢出了,无法处理较大一点的数。所以此处利用杨辉三角求组合数。
杨辉三角的性质:第n行的m个数可表示为 C(n-1,m-1),即为从n-1个不同元素中取m-1个元素的组合数.
代码
#include <iostream>
#include <vector>
#define MOD 1000000007
using namespace std;
long C[105][105];
void init()
{
for