online_judge_1523

最新推荐文章于 2020-03-04 17:29:00 发布

原创最新推荐文章于 2020-03-04 17:29:00 发布 · 333 阅读

0 ·

CC 4.0 BY-SA版权

Online 同时被 2 个专栏收录

262 篇文章

订阅专栏

C++

183 篇文章

订阅专栏

本文介绍如何使用C++创建二叉树并进行层次遍历，包含输入节点数据、连接左右子节点和输出树结构的过程。

#include <iostream>
#include <cstdio>
#include <queue>

using namespace std;

struct BiTNode
{
    int data;
    int left;
    int right;
    bool isRoot;
};

int createTree(BiTNode a[], int n)
{
    int i;
    char c;
    for(i=1; i<=n; ++i)
    {
        scanf("%d",&(a[i].data));
        a[i].left = a[i].right = -1;
        a[i].isRoot = true;
    }
    for(i=1; i<=n; ++i)
    {
        while((getchar()) != '\n');
        scanf("%c",&c);
        if(c == 'd')
        {
            scanf("%d%d",&(a[i].left),&(a[i].right));
            a[a[i].left].isRoot = false;
            a[a[i].right].isRoot = false;
        }
        else if(c == 'l')
        {
            scanf("%d",&(a[i].left));
            a[a[i].left].isRoot = false;
        }
        else if(c == 'r')
        {
            scanf("%d",&(a[i].right));
            a[a[i].right].isRoot = false;
        }
    }
    for(i=1; i<=n; ++i)
    {
        if(a[i].isRoot)
            break;
    }
    return i;
}

void printTree(BiTNode a[], int i)
{
    queue<BiTNode> sq;
    BiTNode bt;
    printf("%d", a[i].data);
    if(a[i].left != -1)
        sq.push(a[a[i].left]);
    if(a[i].right != -1)
        sq.push(a[a[i].right]);
    while(!sq.empty())
    {
        bt = sq.front();
        sq.pop();
        printf(" %d", bt.data);
        if(bt.left != -1)
            sq.push(a[bt.left]);
        if(bt.right != -1)
            sq.push(a[bt.right]);
    }
}

int main()
{
    BiTNode a[1010];
    int n,root;
    while(EOF != scanf("%d", &n))
    {
        root = createTree(a, n);
        printTree(a, root);
        printf("\n");
    }
    return 0;
}

这个题目跟1521题号基本一样的。模块化真好呀！代码可以复用