标签:树的遍历 math ext div ++ code 相等 res algorithm
给定一棵二叉树的后序遍历和中序遍历,请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。
输入第一行给出一个正整数N(≤30),是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。
在一行中输出该树的层序遍历的序列。数字间以1个空格分隔,行首尾不得有多余空格。
7
2 3 1 5 7 6 4
1 2 3 4 5 6 7
#include<cstdio> #include<algorithm> #include<queue> using namespace std; const int maxn = 2e5 + 5; struct Node { int val,l,r; }node[maxn]; int tot = 0; int zhong[maxn],hou[maxn]; int dfs(int len,int zhong[],int hou[]) { if(len <= 0) return -1; if(len == 1) { node[++tot].val = hou[1]; node[tot].l = node[tot].r = -1; // printf("len == 1 %d\n",tot); return tot; } node[++tot].val = hou[len]; // printf("%d %d\n",tot,node[tot].val); int res = tot; int l; for(l = 1;l <= len;l++) if(zhong[l] == hou[len]) break; node[res].l = dfs(l-1,zhong,hou); node[res].r = dfs(len - l,zhong + l,hou + l - 1); return res; } void bfs() { queue<int>que; que.push(1); while(!que.empty()) { int cur = que.front(); que.pop(); if(cur != 1) printf(" "); printf("%d",node[cur].val); if(node[cur].l != -1) que.push(node[cur].l); if(node[cur].r != -1) que.push(node[cur].r); } } int main() { int n; scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d",&hou[i]); for(int i=1;i<=n;i++) scanf("%d",&zhong[i]); dfs(n,zhong,hou); bfs(); return 0; }
标签:树的遍历 math ext div ++ code 相等 res algorithm
原文地址:https://www.cnblogs.com/smallhester/p/10620204.html