1099. Build A Binary Search Tree (30)

最新推荐文章于 2020-06-25 11:32:59 发布

LiuY_ang

最新推荐文章于 2020-06-25 11:32:59 发布

阅读量396

点赞数

CC 4.0 BY-SA版权

分类专栏： PAT 文章标签： PAT

本文链接：https://blog.youkuaiyun.com/LiuY_ang/article/details/70229785

PAT 专栏收录该内容

33 篇文章

订阅专栏

本文介绍了一个PAT-A 1099题目的解决方案，该题目要求构建一棵二叉树并进行层次遍历。文章详细展示了如何通过输入的节点关系和权值构建有序的二叉树，并通过层次遍历输出结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目详情： https://www.patest.cn/contests/pat-a-practise/1099

提交：这里写图片描述

提交代码：

#include <iostream>
#include <algorithm>
using namespace std;
#define N 110
typedef struct Node
{
    int data;
    int leftChild;
    int rightChild;
}Node;
Node node[N]; //nodes[] are numbered from 0 to n-1,and 0 is always the root
int n,weight[N];
int queue[N],front,rear,level[N],top = -1;
int getChildNumber( int root,int &number)  //得到以root为根节点的树中节点的个数，包括该根节点 
{
    if( root != -1 )
    {
        ++number;
        getChildNumber(node[root].leftChild,number);
        getChildNumber(node[root].rightChild,number);
        return number;
    }
    return 0;
}
void build( int root,int *weight )
{
    if( root == -1 ) return ;  //没有节点直接返回 
    int leftNumber = 0,rightNumber = 0;
    leftNumber = getChildNumber(node[root].leftChild,leftNumber);  //得到左子树的节点个数 
    rightNumber = getChildNumber(node[root].rightChild,rightNumber); //得到右子树的节点个数 
    cout<<"leftNumber is "<<leftNumber<<",rightNumber is "<<rightNumber<<endl;
    cout<<"will insert "<<weight[leftNumber]<<endl;9
    node[root].data = weight[leftNumber];
    build(node[root].leftChild,weight); //递归左子树 ，同时改变数组的起始地址，若不改会出现问题 
    build(node[root].rightChild,weight+leftNumber+1); //递归右子树，同时改变数组的起始地址 
}
void levelOrder( int start )
{
    queue[++rear] = start;
    while( front != rear )
    {
        int key = queue[++front];
        if( node[key].leftChild != -1 )
        {
            queue[++rear] = node[key].leftChild;    
        }
        if( node[key].rightChild != -1 )
        {
            queue[++rear] = node[key].rightChild;
        }
        level[++top] = node[key].data;
    }
}
int main ()
{
    cin>>n;
    for( int i=0;i<n;++i )
    {
        int leftKid,rightKid;
        cin>>leftKid>>rightKid;
        node[i].leftChild = leftKid;
        node[i].rightChild = rightKid;
    }
    for( int i=0;i<n;++i )
        cin>>weight[i];
    sort(weight,weight+n); //使weight数组有序 ，以便建立二叉树时使用 
    build(0,weight);       //建立该二叉树 
    rear = front = -1; //层次遍历前的初始化操作 
    levelOrder(0); //进行层次遍历 
    for( int i=0;i<=top;++i ) //输出层次遍历的序列 
    {
        if( i == rear )
            cout<<level[i]<<endl;
        else
            cout<<level[i]<<" ";
    }
    return 0;
}