Fork me on GitHub

PAT(甲级)渡劫(十)-Build A Binary Search(30)

PAT(甲级)渡劫(十)-Build A Binary Search(30)

算法思想:

首先在输入数据的时候就构造二叉树的结构,然后将输入的数据进行排序,将排序好的数字在中序遍历中赋值给构造好的二叉树,最后用层次遍历输出即可。

代码如下:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#define N 200
using namespace std;

struct TNode{
    int data;
    struct TNode *lchild,*rchild;
}Tree[N];
int key[N];
int cnt = 0;
void inOrder(TNode *T){
    if(T != NULL){
        inOrder(T->lchild);
        T->data = key[cnt++];
        inOrder(T->rchild);
    }
}

int main(){
    //freopen("in.txt","r",stdin);
    int n;
    scanf("%d",&n);
    for(int i = 0 ; i <n ; i++){
        int a,b;
        scanf("%d%d",&a,&b);
        if(a != -1){
            Tree[i].lchild = &Tree[a];
        }else{
            Tree[i].lchild = NULL;
        }

        if(b != -1){
            Tree[i].rchild = &Tree[b];
        }else{
            Tree[i].rchild = NULL;
        }
    }
    for(int i = 0 ; i < n ; i++)
        scanf("%d",&key[i]);
    sort(key,key+n);

    inOrder(&Tree[0]);

    queue<TNode> Q;
    Q.push(Tree[0]);
    printf("%d",Tree[0].data);
    while(Q.empty() != true){
        TNode tmp = Q.front();
        Q.pop();
        
        if(tmp.lchild != NULL){
            printf(" %d",tmp.lchild->data);
            Q.push(*(tmp.lchild));
        }
        if(tmp.rchild != NULL){
            printf(" %d",tmp.rchild->data);
            Q.push(*(tmp.rchild));
        }
    }
    return 0;
}

运行结果:

posted on 2019-02-19 14:58  WUSTXIAO  阅读(92)  评论(0编辑  收藏  举报