PTA天梯赛树的复习

Tips

先序遍历：根左右
中序遍历：左根右
后序遍历：左右根

L2-006树的遍历

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

输入格式：
输入第一行给出一个正整数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<iostream>
#include<cstdio>
#include<vector>
#include<queue>
using namespace std;

const int N=40;
struct node{
    int l,r;
}tree[N];
int n,behind[N],mid[N];

int build(int behindL,int behindR,int midL,int midR)//后序遍历子树的左边界，后序遍历子树的右边界，中序遍历子树的左边界，中序遍历子树的右边界
{
    if(behindL>behindR)return 0;
    int rt=behind[behindR];//后序遍历的父结点在整个子树最后
    int p1=midL;//从midL开始
    while(mid[p1]!=rt)++p1;//找中序遍历的父结点下标
    int p2=p1-midL;//p1从midL开始，减去midL等于左子树的个数
    tree[rt].l=build(behindL,behindL+p2-1,midL,p1-1);
    tree[rt].r=build(behindL+p2,behindR-1,p1+1,midR);
    return rt;
}
void bfs(int rt)
{
    vector<int>vec;
    queue<int>q;
    q.push(rt);
    while(!q.empty())
    {
        rt=q.front();
        q.pop();
        vec.push_back(rt);
        if(tree[rt].l!=0)
            q.push(tree[rt].l);
        if(tree[rt].r!=0)
            q.push(tree[rt].r);
    }
    for(int i=0;i<n-1;++i)
        printf("%d ",vec[i]);
    printf("%d\n",vec.back());
}
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=0;i<n;++i)
            scanf("%d",&behind[i]);
        for(int i=0;i<n;++i)
            scanf("%d",&mid[i]);
        int rt=build(0,n-1,0,n-1);
        bfs(rt);
    }
    return 0;
}

L2-011玩转二叉树

给定一棵二叉树的中序遍历和前序遍历，请你先将树做个镜面反转，再输出反转后的层序遍历的序列。所谓镜面反转，是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。

输入格式：
输入第一行给出一个正整数N（≤30），是二叉树中结点的个数。第二行给出其中序遍历序列。第三行给出其前序遍历序列。数字间以空格分隔。

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

输入样例：

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

输出样例：

4 6 1 7 5 3 2

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
using namespace std;

const int N=40;
struct node{
    int l,r;
}tree[N];
int n,pre[N],mid[N];

int build(int preL,int preR,int midL,int midR)//前序遍历子树的左边界，前序遍历子树的右边界，中序遍历子树的左边界，中序遍历子树的右边界
{
    if(preL>preR)return 0;
    int rt=pre[preL];//前序遍历的父结点在整个子树开头
    int p1=midL;//从midL开始
    while(mid[p1]!=rt)++p1;//找中序遍历的父结点下标
    int p2=p1-midL;//p1从midL开始，减去midL等于左子树的个数
    tree[rt].l=build(preL+1,preL+p2,midL,p1-1);
    tree[rt].r=build(preL+p2+1,preR,p1+1,midR);
    return rt;
}
void bfs(int rt)
{
    vector<int>vec;
    queue<int>q;
    q.push(rt);
    while(!q.empty())
    {
        rt=q.front();
        q.pop();
        vec.push_back(rt);
        if(tree[rt].r!=0)
            q.push(tree[rt].r);
        if(tree[rt].l!=0)
            q.push(tree[rt].l);
    }
    for(int i=0;i<n-1;++i)
        printf("%d ",vec[i]);
    printf("%d\n",vec.back());
}
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=0;i<n;++i)
            scanf("%d",&mid[i]);
        for(int i=0;i<n;++i)
            scanf("%d",&pre[i]);
        int rt=build(0,n-1,0,n-1);
        bfs(rt);
    }
    return 0;
}

L2-035完全二叉树的层序遍历

一个二叉树，如果每一个层的结点数都达到最大值，则这个二叉树就是完美二叉树。对于深度为 D 的，有 N 个结点的二叉树，若其结点对应于相同深度完美二叉树的层序遍历的前 N 个结点，这样的树就是完全二叉树。

给定一棵完全二叉树的后序遍历，请你给出这棵树的层序遍历结果。

输入格式：
输入在第一行中给出正整数 N（≤30），即树中结点个数。第二行给出后序遍历序列，为 N 个不超过 100 的正整数。同一行中所有数字都以空格分隔。

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

输入样例：

8
91 71 2 34 10 15 55 18

输出样例：

18 34 55 71 2 10 15 91

#include<iostream>
#include<cstdio>
using namespace std;

const int N=40;
int n,len,behind[N],ans[N];

void build(int x)
{
    if(x>=n)return;
    build(2*x+1);
    build(2*x+2);
    ans[x]=behind[len++];
}
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=0;i<n;++i)
            scanf("%d",&behind[i]);
        build(0);
        for(int i=0;i<n-1;++i)
            printf("%d ",ans[i]);
        printf("%d\n",ans[n-1]);
    }
    return 0;
}

L3-010是否完全二叉搜索树

将一系列给定数字顺序插入一个初始为空的二叉搜索树（定义为左子树键值大，右子树键值小），你需要判断最后的树是否一棵完全二叉树，并且给出其层序遍历的结果。

输入格式：
输入第一行给出一个不超过20的正整数N；第二行给出N个互不相同的正整数，其间以空格分隔。

输出格式：
将输入的N个正整数顺序插入一个初始为空的二叉搜索树。在第一行中输出结果树的层序遍历结果，数字间以1个空格分隔，行的首尾不得有多余空格。第二行输出YES，如果该树是完全二叉树；否则输出NO。

输入样例1：

9
38 45 42 24 58 30 67 12 51

输出样例1：

38 45 24 58 42 30 12 67 51
YES

输入样例2：

8
38 24 12 45 58 67 42 51

输出样例2：

38 45 24 58 42 12 67 51
NO

#include<iostream>
#include<cstdio>
#include<queue>
#include<vector>
using namespace std;

struct node{
    int l,r,x;
}tree[30];
int idx,MAX;
vector<int>v;

void insert(int&rt,int x,int sum)
{
    if(rt==-1)
    {
        tree[idx].l=-1;
        tree[idx].r=-1;
        tree[idx].x=x;
        rt=idx++;
        MAX=max(MAX,sum);
        return;
    }
    if(x>tree[rt].x)
        insert(tree[rt].l,x,sum*2);
    else
        insert(tree[rt].r,x,sum*2+1);
}
void bfs(int rt)
{
    queue<int>q;
    q.push(rt);
    while(!q.empty())
    {
        rt=q.front();
        q.pop();
        v.push_back(tree[rt].x);
        if(tree[rt].l!=-1)
            q.push(tree[rt].l);
        if(tree[rt].r!=-1)
            q.push(tree[rt].r);
    }
}
int main()
{
    int n,x,rt=-1;
    scanf("%d",&n);
    for(int i=0;i<n;++i)
    {
        scanf("%d",&x);
        insert(rt,x,1);
    }
    bfs(rt);
    for(int i=0;i<n-1;++i)
        printf("%d ",v[i]);
    printf("%d\n",v.back());
    if(MAX==n)puts("YES");
    else puts("NO");
    return 0;
}