【PAT甲级 二叉排序树】1099 Build A Binary Search Tree (30 分)

# include <bits/stdc++.h>
using namespace std;

int N;
struct Node{
    
    
    int data;
    int lchild, rchild;
    Node(int d, int l, int r):data(d), lchild(l), rchild(r){
    
    }
    //  没事写一下C++11的委托构造函数
    Node(            ):Node(0, -1, -1){
    
    }
    Node(int d       ):Node(d, -1, -1){
    
    }
    Node(int l, int r):Node(0,  l,  r){
    
    }

}node[100];
int num[100];

int idx = 0;
void inOrder(int root){
    
     // 中序遍历给结点赋值
    if(root == -1) return;
    inOrder(node[root].lchild);
    node[root].data = num[idx++];
    inOrder(node[root].rchild);
}

int cnt = 0;
void levelOrder(){
    
     // 层序遍历输出结果
    queue<int> q;
    q.push(0);
    while(!q.empty()){
    
    
        int u = q.front();
        q.pop();
        cnt++;
        cout << node[u].data << (cnt == N ? "\n" : " ");

        if(node[u].lchild != -1) q.push(node[u].lchild);
        if(node[u].rchild != -1) q.push(node[u].rchild);
    }
    
}

int main(){
    
    
    cin >> N;
    for(int i = 0;i < N;++i)
        cin >> node[i].lchild >> node[i].rchild;
    for(int i = 0;i < N;++i)
        cin >> num[i];
    sort(num, num+N);
    inOrder(0);
    levelOrder();

    return 0;
}

猜你喜欢

转载自blog.csdn.net/MYMarcoreus/article/details/113945922