PAT (Advanced Level) Practice - 1127 ZigZagging on a Tree（30 分）

树的“S”形层序遍历解题思路

最新推荐文章于 2021-03-09 11:40:09 发布

原创最新推荐文章于 2021-03-09 11:40:09 发布 · 391 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#PAT (Advanced Level) Practice #1127 ZigZagging on a Tree 30 分

PTA 专栏收录该内容

173 篇文章

订阅专栏

博客给出题目链接，题目要求根据中序和后序序列，按“S”形层序遍历输出。解题思路是用queue保存结点，遍历该层所有结点后将queue里的结点弹到stack里，同时要确定每次从哪个头开始弹入queue，还给出了AC代码。

题目链接：点击打开链接

题目大意：给出中序和后序序列，按照“S”形层序遍历输出。

解题思路：找规律可以发现，只要每次用queue保存，然后遍历了该层的所有结点后，把queue里的结点弹到stack里即可。注意的是，这里需要每次从哪个头开始弹入queue里，需要想好是先 l,r or r,l。

AC 代码

#include<bits/stdc++.h>
#include<cmath>

#define mem(a,b) memset(a,b,sizeof a)
#define ssclr(ss) ss.clear(), ss.str("")
#define INF 0x3f3f3f3f
#define MOD 1000000007

using namespace std;

typedef long long ll;

struct node
{
    int d,l,r;
}T[40];

int n;
int in[40], post[40], lvl[40];

int create(int l1,int r1, int l2,int r2, int l) // in post
{
    if(l1>r1) return -1;
    lvl[l]++;
    int rt=r2;
    int p1=l1,p2;

    while(in[p1]!=post[r2]) p1++;
    p2=p1-l1;
    T[rt].d=post[rt];
    T[rt].l=create(l1,p1-1,l2,l2+p2-1,l+1);
    T[rt].r=create(p1+1,r1,l2+p2,r2-1,l+1);

    return rt;
}

void bfs(int rt)
{
    queue<int> q;
    vector<int> v;
    stack<int> sk;
    q.push(rt);
    node nd;
    int th=0, cnt=0;

    while(v.size()<n)
    {
        if(!sk.empty()) rt=sk.top(), sk.pop();
        else if(!q.empty()) rt=q.front(), q.pop();
        v.push_back(T[rt].d), cnt++;

        if(th%2==0)
        {
            if(T[rt].r!=-1) q.push(T[rt].r);
            if(T[rt].l!=-1) q.push(T[rt].l);
        }
        else
        {
            if(T[rt].l!=-1) q.push(T[rt].l);
            if(T[rt].r!=-1) q.push(T[rt].r);
        }

        if(lvl[th]==cnt)
        {
            cnt=0, th++;
            while(!q.empty()) sk.push(q.front()), q.pop();
        }
    }

    for(int i=0;i<v.size();i++) printf("%d%c",v[i],i==v.size()-1?'\n':' ');
}

int main()
{
    int rt;
    scanf("%d",&n);
    for(int i=0;i<n;i++) scanf("%d",&in[i]);
    for(int i=0;i<n;i++) scanf("%d",&post[i]);
    rt=create(0,n-1,0,n-1,0);
    bfs(rt);

    return 0;
}