PAT 1127ZigZagging on a Tree

最新推荐文章于 2020-10-29 15:19:46 发布

drawAgirl

最新推荐文章于 2020-10-29 15:19:46 发布

阅读量251

点赞数

分类专栏： PAT

本文链接：https://blog.youkuaiyun.com/drawAgirl/article/details/85646698

版权

PAT 专栏收录该内容

19 篇文章

订阅专栏

本文介绍了一种使用C++实现的二叉树遍历算法，重点在于根据给定的中序和后序遍历序列构建二叉树，并通过层次遍历来输出各层节点值。文章详细解释了如何递归地构建二叉树，以及如何通过队列实现层次遍历，同时展示了如何根据遍历方向调整输出顺序。

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

leetcode上有一题和这个基本一样，感觉这题有点水。


#include <cstdio>
#include <iostream>
#include <cmath>
#include <algorithm>
#include <vector>
#include <queue>

using namespace std;

typedef struct node {
    int val;
    struct node *left;
    struct node *right;
}Node,*TreeNode;
vector<int> post;
vector<int> in;
vector<int> ans;
vector<vector<int>> vv;
TreeNode complete(int inL,int inR,int postL,int postR){
    if (inL > inR) {
        return NULL;
    }
    TreeNode node = new Node;
    node->val = post[postR];
    node->left = node->right = NULL;
    int i;
    for (i=inL; i<inR;i++) {
        if (in[i] == post[postR]) {
            break;
        }
    }
    int numleft = i - inL;
    node->left = complete(inL, i-1, postL, postL+numleft-1);
    node->right = complete(i+1, inR, postL+numleft, postR-1);
    return node;
}

void level(TreeNode root){
    int flag = 1;
    queue<TreeNode> q;
    q.push(root);
    while (!q.empty()) {
        int size = q.size();
        ans.clear();
        while (size--) {
            TreeNode node = q.front();
            q.pop();
            if (node->left) q.push(node->left);
            if (node->right) q.push(node->right);
            ans.push_back(node->val);
        }
        flag *= -1;
        if (flag == -1) {
            reverse(ans.begin(), ans.end());
        }
        vv.push_back(ans);
    }
}


int main(){
    int n;
    scanf("%d",&n);

    for (int i=0; i<n; i++) {
        int a;
        scanf("%d",&a);
        in.push_back(a);
    }
    for (int i=0; i<n; i++) {
        int a;
        scanf("%d",&a);
        post.push_back(a);
    }
    TreeNode root = complete(0, n-1, 0, n-1);
    int space = 0;
    level(root);
    for (int i=0; i<vv.size(); i++) {
        for (int j=0; j<vv[i].size(); j++) {
            printf("%d",vv[i][j]);
            space++;
            if (space != n) {
                printf(" ");
            }
        }
    }
}