PTA-二叉搜索树的插入排序

醉酒当歌 提交于 2019-12-29 20:38:27

二叉搜索树的插入排序

二叉搜索树定义为:

左子树仅包含小于根结点的元素;
右子树仅包含大于根结点的元素;
左右子树均为二叉搜索树。
一棵二叉搜索树可以有不同的插入顺序。例如,对于如下的二叉搜索树
在这里插入图片描述
figure.png

插入序列可以是 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

解题思路

1、先构造杨辉三角(用于存放组合数)。
2、先根据输入的序列先序构造成一个树。
3、根据这个已经构造好的二叉树计算它的插入序列总数,具体计算方式如下
对于一个结点
1、如果是叶子结点,那么它的插入序列只有一种
2、若果只有左子树,那么它的插入序列数和他的左子树一样
3、如果只有右子树,那么它的插入序列数和他的右子树一样
4、如果左右子树都有,则它的插入序列数等于左子树的序列个数右子树的序列个数
(组合数)C(左子树的元素个数+右子树的元素个数)(左子树元素个数)

代码

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
typedef struct node{ //结点结构
    int data; //结点数据
    long n; //当前结点作为根结点时的序列种类
    int num; //当前结点的子树元素的个数
    struct node * lchild; //左孩子
    struct node * rchild; //右孩子
}node,* tree;
tree creattree(int *p,int n){ //先序构建二叉树,P为输入序列首地址,n为元素个数
    tree t; //声明一个结点空间
    if(n<=0||p[0]==-1) //n=0或到达-1则标志已经结束
        return NULL;
    t=(tree)malloc(sizeof(node)); //分配空间
    t->data=p[0]; //当前结点的数据域赋值
    t->n=1; //当前只有一种
    t->num=n; //当前字树元素个数
    int flag; //设立一个标志,用于记录大于根结点的下一个结点的上一个元素的下标
    for(int i=1;i<=n-1;i++){ //循环找出大于根结点的下一个结点的上一个元素的下标
        if(p[i]<p[0]){ //找出小于根结点的元素
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
typedef struct node{ //结点结构
    int data; //结点数据
    long long n; //当前结点作为根结点时的序列种类
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
typedef struct node{ //结点结构
    int data; //结点数据
    long long n; //当前结点作为根结点时的序列种类
    int num; //当前结点的子树元素的个数
    struct node * lchild; //左孩子
    struct node * rchild; //右孩子
}node,* tree;
tree creattree(int *p,int n){ //先序构建二叉树,P为输入序列首地址,n为元素个数
    tree t; //声明一个结点空间
    if(n<=0||p[0]==-1) //n=0或到达-1则标志已经结束
        return NULL;
    t=(tree)malloc(sizeof(node)); //分配空间
    t->data=p[0]; //当前结点的数据域赋值
    t->n=1; //当前只有一种序列
    t->num=n; //当前子树元素个数
    int flag; //设立一个标志,用于记录大于根结点的下一个结点的上一个元素的下标
    for(int i=1;;i++){ //循环找出大于根结点的下一个结点下标
        if(p[i]>p[0]){ //找出大于根结点的元素的下标
            flag=i;
            break;
        }
    }
    t->lchild=creattree(p+1,flag-1);//先序情况下从根结点的下一个结点开始,到大于根结点的结点之前,都是其左子树
    t->rchild=creattree(p+flag,n-flag); //先序情况下大于根结点的结点到结束,都是右子树
    return t; //返回根结点
}
int yanghui[101][101]; //构建杨辉矩阵
void creatyanghui(){ //构建杨辉三角
    for (int i=0;i<=100;i++){
        yanghui[i][0]=1;
        yanghui[i][i]=1;
        for (int j=1;j<i;j++) {
            yanghui[i][j]=yanghui[i-1][j]+yanghui[i-1][j-1];
            yanghui[i][j]%=1000000007; //随着循环,到后面yanghui[i][j]的值已经极其庞大,远远超过了int的界限,这里先取余数
        }
    }
}
void count(tree t){ //后续遍历,添加序列种类
    if(t){ //非空树
        count(t->lchild);
        count(t->rchild);
        if(!(t->lchild)&&!(t->rchild)) //叶子结点,只有一种
            t->n=1;
        else if((t->lchild)&&!(t->rchild)) //只有左子树,种类和左子树一样
            t->n=t->lchild->n;
        else if((t->rchild)&&!(t->lchild)) //只有右子树,种类和右子树一样
            t->n=t->rchild->n;
        else //左右子树都有
            t->n=(t->lchild->n)%1000000007*(t->rchild->n)*yanghui[t->lchild->num+t->rchild->num][t->lchild->num];
        //同杨辉矩阵一样,到后面t->n的值已经极其庞大,所以这里每次计算都对1000000007取余数,防止越界,而对结果又不影响
    }
};
int main(void){
    creatyanghui();
    int gets[101]; //结点序列
    int num; //结点个数
    scanf("%d",&num); //输入结点个数
    for(int i=0;i<=num-1;i++){ //输入结点序列
        scanf("%d",&gets[i]);
        if(i==num-1) gets[i+1]=-1; //标志序列结束
    }
    tree t; //声明二叉树
    t=creattree(gets,num); //构建先序二叉树
    count(t);
    printf("%d",(t->n));
}

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!