PAT A1127 ZigZagging on a Tree (30 分) 二叉树遍历

最新推荐文章于 2024-11-07 20:09:45 发布

原创最新推荐文章于 2024-11-07 20:09:45 发布 · 290 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

PAT

该博客介绍了如何根据中序和后序序列还原二叉树，并实现一种特殊的ZigZag层序遍历。在遍历过程中，奇数层节点从左到右输出，偶数层节点从右到左输出。提供了AC代码作为解决方案。

题目大意：给出中序和后序序列，还原二叉树，然后进行一种Zigzag的层序遍历，即从根节点(记为第0层)开始输出，第1层从左向右输出节点，第2层从右向左输出节点...依次循环。

根据中序和后序序列还原二叉树。然后就是进行普通的层序遍历，并且更新每个节点的层数信息layer域。同时因为节点数不超过30，开一个大小为40的vector[40]数组，用于记录每一层遍历的结果。最后，如果是奇数层，直接输出vector[k]，如果是偶数层，就倒序输出vector[k]。

AC代码：

#include <vector>
#include <cstdio>
#include <queue>
using namespace std;

struct Node
{
    int data;
    int level;
    Node* left;
    Node* right;
    Node(int data):data(data), left(NULL), right(NULL){};
};

vector<int> in, post;
vector<vector<int>> level(40);

void creatFromInPost(Node* &node, int inl, int inr, int postl, int postr)
{
    if(postl > postr) return;
    if(node == NULL) node = new Node(post[postr]);
    int index = inl;
    for(; index <= inr; index++)
    {
        if(in[index] == post[postr]) break;
    }
    int leftNum = index - inl;
    creatFromInPost(node->left, inl, index - 1, postl, postl + leftNum - 1);
    creatFromInPost(node->right, index + 1, inr, postl + leftNum, postr - 1);
}

void levelOrder(Node* node, int &maxLevel)
{
    node->level = 0;
    queue<Node*> q;
    q.push(node);
    while(!q.empty())
    {
        Node* now = q.front();
        q.pop();
        if(now->level > maxLevel) maxLevel = now->level;
        level[now->level].push_back(now->data);
        if(now->left)
        {
            now->left->level = now->level + 1;
            q.push(now->left);
        }
        if(now->right)
        {
            now->right->level = now->level + 1;
            q.push(now->right);
        }
    }
}

int main()
{
    int N;
    scanf("%d", &N);
    in.resize(N);
    post.resize(N);
    for (int i = 0; i < N; ++i)
        scanf("%d", &in[i]);
    for (int i = 0; i < N; ++i)
        scanf("%d", &post[i]);
    Node* root = NULL;
    creatFromInPost(root, 0, N-1, 0, N-1);
    int maxLevel = 0;
    levelOrder(root, maxLevel);
    printf("%d", root->data);
    for (int i = 1; i <= maxLevel; ++i)
    {
        if(i % 2 == 1)
        {
            for (int j = 0; j < level[i].size(); ++j)
                printf(" %d", level[i][j]);
        }
        else
        {
            for (int j = level[i].size() - 1; j >= 0; --j)
                printf(" %d", level[i][j]);
        }
    }
    return 0;
}