PTA二叉树习题（二叉树翻转）

链接: link.
这道题主要就是考察二叉树的重建和翻转，也就是左右子树互换。我们直接看代码。

#include <bits/stdc++.h>

using namespace std;

const int N = 35;
int str1[N],str2[N];

struct node
{
    int data;
    struct node *l,*r;
};

struct node *create(int str1[],int str2[],int n)
{//根据先序遍历和中序遍历重建二叉树
    if(n==0) return NULL;
    struct node *root;
    root = new node;
    root->data = str1[0];
    int i;
    for(i=0;i<n;i++)
    {
        if(str2[i]==str1[0])
            break;
    }
    root->l = create(str1+1,str2,i);
    root->r = create(str1+i+1,str2+i+1,n-i-1);
    return root;
};

struct node *Move(struct node *root)
{
    if(!root) return NULL;
    //递归分别处理左右子树，注意要先进行这一步
    root->l = Move(root->l);
    root->r = Move(root->r);
    //类似于数值的交换
    struct node *t = root->l;
    root->l = root->r;
    root->r = t;
    return root;
};

//层次遍历，level储存结果
vector<int> level;
void levelorder(struct node *root)
{
    if(!root) return ;
    queue<node*> que;
    que.push(root);
    while(!que.empty())
    {
        struct node *p;
        p = que.front();
        que.pop();
        level.push_back(p->data);
        if(p->l) que.push(p->l);
        if(p->r) que.push(p->r);
    }
}

int main()
{
    int n;
    cin>>n;
    for(int i=0;i<n;i++) cin>>str2[i];
    for(int i=0;i<n;i++) cin>>str1[i];
    struct node *root;
    root = create(str1,str2,n);
    root = Move(root);
    levelorder(root);
    for(int i=0;i<level.size();i++)
    {
        if(i!=0) cout<<' ';
        cout<<level[i];
    }
    return 0;
}

链接: link.
这个题较上一个题麻烦一点，不过也还是考察重建二叉树和二叉树的翻转。我们还需要判断题目给到的序列是二叉树翻转前还是翻转后的前序遍历序列，并输出后序遍历序列。

#include <bits/stdc++.h>

using namespace std;

const int N = 1010;

struct node
{
    int data;
    struct node *l,*r;
};

//重建二叉树
struct node *Insert(struct node *root,int data)
{
    if(!root)
    {
        root = new node;
        root->data = data;
        root->l = NULL;
        root->r = NULL;
    }
    else
    {
        if(data<root->data)
            root->l = Insert(root->l,data);
        else
            root->r = Insert(root->r,data);
    }
    return root;
};

//先序遍历
vector<int> pre;
void preorder(struct node *root)
{
    if(!root) return ;
    pre.push_back(root->data);
    preorder(root->l);
    preorder(root->r);
}

//后序遍历
vector<int> post;
void postorder(struct node *root)
{
    if(!root) return ;
    postorder(root->l);
    postorder(root->r);
    post.push_back(root->data);
}

//翻转
struct node *Move(struct node *root)
{
    if(!root) return NULL;
    root->l = Move(root->l);
    root->r = Move(root->r);
    struct node *t = root->l;
    root->l = root->r;
    root->r = t;
    return root;
};

int main()
{
    int n;
    cin>>n;
    struct node *root = NULL;
    vector<int> num;
    for(int i=0;i<n;i++)
    {
        int x;
        cin>>x;
        root = Insert(root,x);
        num.push_back(x);
    }
    preorder(root);
    bool flag = true;
    //判断是否是二叉排序树的前序遍历
    for(int i=0;i<n;i++)
    {
        if(num[i]!=pre[i])
        {
            flag = false;
            break;
        }
    }
    if(flag)//如果是，那么直接输出结果
    {
        cout<<"YES"<<endl;
        postorder(root);
        for(int i=0;i<n;i++)
        {
            if(i!=0)
                cout<<" ";
            cout<<post[i];
        }
        return 0;
    }
    //如果不是，我们进行翻转二叉树操作，再次求得先序遍历序列
    root = Move(root);
    pre.clear();//注意要清空一下
    preorder(root);
    flag = true;
    for(int i=0;i<n;i++)
    {
        if(num[i]!=pre[i])
        {
            flag = false;
            break;
        }
    }
    if(flag)//如果是，输出结果
    {
        cout<<"YES"<<endl;
        postorder(root);
        for(int i=0;i<n;i++)
        {
            if(i!=0) cout<<' ';
            cout<<post[i];
        }
        return 0;
    }
    //如果不是，就输出NO
    cout<<"NO"<<endl;
    return 0;
}