码迷,mamicode.com
首页 > 其他好文 > 详细

【7-10 PAT】树的遍历

时间:2019-03-29 11:44:18      阅读:144      评论:0      收藏:0      [点我收藏+]

标签:树的遍历   math   ext   div   ++   code   相等   res   algorithm   

给定一棵二叉树的后序遍历和中序遍历,请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。

输入格式:

输入第一行给出一个正整数N(30),是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。

输出格式:

在一行中输出该树的层序遍历的序列。数字间以1个空格分隔,行首尾不得有多余空格。

输入样例:

7
2 3 1 5 7 6 4
1 2 3 4 5 6 7

输出样例:

4 1 6 3 5 7 2

 

 

#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;
}

 

【7-10 PAT】树的遍历

标签:树的遍历   math   ext   div   ++   code   相等   res   algorithm   

原文地址:https://www.cnblogs.com/smallhester/p/10620204.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!