18版考研数据结构天勤课后习题代码-二叉树 重难点代码

1、

区分insert函数构造二叉排序树、由中后序构造树、由先序构造满二叉树的区别

insert函数构造二叉排序树：是一个节点一个节点插入

由中后序构造树、由先序构造满二叉树：通过递归把一个数组构建成一棵树


/*
//查找值为key的节点 天勤P140 例6-3

BiTree *insert(BiTree *&root,char ch)
{
    if(root==NULL)
    {
        root=(BiTree*)malloc(sizeof(BiTree));
        root->data=ch;
        root->lchild=NULL;
        root->rchild=NULL;
        return root;
    }
    else if(ch<=root->data)
    {
        root->lchild=insert(root->lchild,ch);
    }
    else if(ch>root->data)
    {
        root->rchild=insert(root->rchild,ch);
    }
    return root;
}
void preOrder(BiTree *t,int k)
{
    stack<BiTree*> s;
    int x=0;
    while(t!=NULL||!s.empty())
    {
        while(t)
        {
            s.push(t);
            x++;
            if(x==k)
            {
                cout<<t->data;
            }
            t=t->lchild;
        }
        if(!s.empty())
        {
            t=s.top();
            s.pop();
            t=t->rchild;
        }
    }
}
int main()
{
    BiTree *root=NULL;
    int n;
    char a[101];
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>a[i];
        root=insert(root,a[i]);
    }
    preOrder(root,2);
}
*/


/*
//由前序序列和中序序列构造二叉树  天勤P162 真题仿造
BiTree *CreateTree(char pre[],char in[],int pre1,int pre2,int in1,int in2)
{
    if(pre1>pre2)  //注意跳出条件，注意和二叉树的insert函数区分，insert函数只是插入一个节点，而此函数是根据一个数组建一棵树，因此建完节点不用 return，越界时才return
    {
        return NULL;
    }
        BiTree *t=(BiTree*)malloc(sizeof(BiTree));
        t->data=pre[pre1];
        t->lchild=t->rchild=NULL;
    int k;
    for(int i=in1;i<=in2;i++)
    {
        if(in[i]==pre[pre1])
        {
            k=i;
            break;
        }
    }
    t->lchild=CreateTree(pre,in,pre1+1,pre1+k-in1,in1,k-1);  //易错点！！！注意下标不要写错了
 
    t->lchild=CreateTree(t->lchild,pre,in,pre1+1,pre1+k-in1,in1,k-1);
 
    t->rchild=CreateTree(pre,in,pre1+k-in1+1,pre2,k+1,in2);
    return t;
}
 
void postOrder(BiTree *t)
{
    if(t!=NULL)
    {
        postOrder(t->lchild);
        postOrder(t->rchild);
        cout<<t->data<<endl;
    }
}
int main()
{
    char pre[101];
    char in[101];
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>pre[i];
    }
    for(int i=1;i<=n;i++)
    {
        cin>>in[i];
    }
    BiTree *root=NULL;
    root=CreateTree(pre,in,1,n,1,n);
    postOrder(root);
    
}
*/


//满二叉树先序序列构造树，然后写出后序序列 天勤P165 （二）（6）
BiTree *CreateTree(char a[],int i,int n)
{
    BiTree *t;
    if (i>n)
    {
        return NULL;
    }
    t=(BiTree*)malloc(sizeof(BiTree));
    t->data=a[i];
    t->lchild=t->rchild=NULL;
    t->lchild=CreateTree(a,2*i,n);
    t->rchild=CreateTree(a,2*i+1,n);
    return t;
}

void postOrder(BiTree *t)
{
   if(t)
   {
       postOrder(t->lchild);
       postOrder(t->rchild);
       cout<<t->data<<" ";
   }
}

int main()
{
    int n;
    cin>>n;
    char a[101];
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    BiTree *root=NULL;
    root=CreateTree(a,1,n);
    postOrder(root);
}

2、求层数 增加一个存放层数的变量

//求二叉树的宽度  天勤P144 例6-5
typedef struct BiTree
{
    char data;
    int lev;   //存放节点的层数
    struct BiTree *lchild,*rchild;
    
}BiTree;
BiTree *insert(BiTree *&root,char ch)
{
    if(root==NULL)
    {
        root=(BiTree*)malloc(sizeof(BiTree));
        root->data=ch;
        root->lchild=NULL;
        root->rchild=NULL;
        return root;
    }
    else if(ch<=root->data)
    {
        root->lchild=insert(root->lchild,ch);
    }
    else if(ch>root->data)
    {
        root->rchild=insert(root->rchild,ch);
    }
    return root;
}

int Width(BiTree *t)
{
    BiTree *q[maxSize];
    int front=1;
    int rear=1;
    int l;
    if(t!=NULL)  //不要写成while
    {
        t->lev=1;   //判断完了再入队
        q[rear]=t;
        rear++;
        while(rear!=front)
      {
        BiTree *s=q[front];
        l=s->lev;
        front++;
        if(s->lchild!=NULL)
        {
            s->lchild->lev=l+1;
            q[rear]=s->lchild;
            rear++;
        }
        if(s->rchild!=NULL)
        {
            s->rchild->lev=l+1;
            q[rear]=s->rchild;
            rear++;
        }
      }
    }
    int max=0;
    for(int i=1;i<=l;i++)
    {
        int t=0;
        for(int j=1;j<rear;j++)
        {
            if(q[j]->lev==i)
            {
                t++;
            }
        }
        if(t>max)
        {
            max=t;
        }
    }
    return max;
  }

int main()
{
    BiTree *root=NULL;
    int n;
    char a[101];
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>a[i];
        root=insert(root,a[i]);
    }
    //inOrder(root);
    int d=Width(root);
    cout<<d<<endl;
}
*/