1127. ZigZagging on a Tree (30)

简介: #include #include #include using namespace std;int n;const int maxn = 31;struct node { int data; node *l,...
#include <iostream>
#include <vector>
#include <queue>
using namespace std;

int n;
const int maxn = 31;
struct node { int data; node *l, *r;};
vector<int> post(maxn), in(maxn), ans, lnum(maxn);

node* build(int postL, int postR, int inL, int inR){//根据后序和中序建树
    if(postL > postR) return NULL;
    node *root = new node;
    root->data = post[postR];
    int k;
    for (k = inL; k <= inR; k++)
        if(in[k] == post[postR]) break;
    int numLeft = k - inL;
    root->l = build(postL, postL + numLeft - 1, inL, k - 1);
    root->r = build(postL + numLeft, postR - 1, k + 1, inR);
    return root;
}
void dfs(node *root, int depth){//求每一层的个数
    if (root == NULL) return;
    lnum[depth]++;
    if(root->l != NULL) dfs(root->l, depth + 1);
    if(root->r != NULL) dfs(root->r, depth + 1);
}
void bfs(node *root){//层序遍历的结果存在ans中
    queue<node*> q;
    q.push(root);
    while (!q.empty()) {
        node *now = q.front();
        q.pop();
        ans.push_back(now->data);
        if(now->l) q.push(now->l);
        if(now->r) q.push(now->r);
    }
}

int main(){
    cin >> n;
    for(int i = 0; i < n; i++) cin >> in[i];
    for(int i = 0; i < n; i++) cin >> post[i];
    node *root = build(0, n - 1, 0, n - 1);
    dfs(root, 0);
    bfs(root);
    //z输出
    int cnt = 0;
    printf("%d", ans[cnt++]);
    for (int i = 1; i < maxn;) {
        if (i % 2) {
            for (int j = 0; j < lnum[i]; j++)
                printf(" %d", ans[cnt+j]);
            cnt = cnt + lnum[i];
            i++;
        }else{
            for (int j = lnum[i] - 1; j >= 0; j--)
                printf(" %d", ans[cnt+j]);
            cnt = cnt + lnum[i];
            i++;
        }
    }
    cout << endl;

    return 0;
}
目录
相关文章
|
6月前
|
缓存 索引
图解B Tree和B+ Tree
图解B Tree和B+ Tree
66 0
|
Python
二叉搜索树(Binary Search Tree
二叉搜索树(Binary Search Tree,简称 BST)是一种特殊的二叉树结构,它的每个节点具有以下性质:
78 4
|
5月前
|
存储 算法 编译器
|
6月前
|
定位技术 索引
R-tree 总结
R-tree 总结
78 0
|
6月前
|
存储 算法 Python
赢者树(Losers Tree)
赢者树(Losers Tree)是一种经典的数据结构,常用于外部排序(External Sorting)算法中,将多个有序的子序列合并成一个有序的序列。赢者树本质上是一棵完全二叉树,每个节点存储着一个子序列的最小值。每次合并操作时,比较各个子序列的最小值,选出最小值并将其存入输出序列中,同时将该最小值所在的节点从赢者树中删除,并将其对应的子序列的下一个元素作为新的最小值插入到赢者树中进行调整,直到所有子序列的元素都被合并完成。
68 3
树(Tree)和二叉树(Binary Tree)——(代码篇)
树(Tree)和二叉树(Binary Tree)——(代码篇)
75 0
|
存储 分布式数据库
树(Tree)和二叉树(Binary Tree)——(概念篇)
树(Tree)和二叉树(Binary Tree)——(概念篇)
68 0
|
存储 数据格式
1367:查找二叉树(tree_a)
1367:查找二叉树(tree_a)
|
存储 数据库 索引
B-Tree和B+Tree特点
B - Tree和B + Tree特点
138 0
|
数据库 索引
B-Tree, B+Tree
B-Tree, B+Tree
83 0