1102 Invert a Binary Tree (25分)[二叉树][中序遍历][层序遍历][树反转]

By Jalan

知识工具需求

数学

数据结构和算法

语言

题干

反转一颗二叉树

输入条件

0-N-1的N<=10个节点,所有节点都在数里.
下面是按编号的每个节点的子,没有是-

输出条件

第一行输出层序,第二行输出反转后的中序,末尾无空格
注意层序是右到左

例子

例1

输入

8
1 -
- -
0 -
2 7
- -
- -
5 -
4 6

输出

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

题解

第一次

思路

在给的数据里应该每个子节点都出现过,则没有出现过的节点就是根节点.根据根节点和儿子表就可以很轻松的得到层序.
按照右根左的方式遍历树就可以输出反转后的中序.
所以其实重点是怎么遍历这个特殊的数结构.

• 建树也许可以?我没试
• 但是感觉也没必要.写个栈就行.

1. 输入
2. 找根.
3. 写个队列层序一下,输出
4. 写个栈中序一下,输出

预期时间复杂度

n

编写用时

35分钟

代码

CPP

#include <iostream>
#include <queue>
#include <stack>
#include <stdio.h>
#include <string>
#include <vector>
using namespace std;
struct node
{
    int l = -2;
    int r = -2;
};

int main(int argc, char const *argv[])
{
    //1
    int N;
    scanf("%d", &N);
    vector<node> t(N);
    vector<int> r(N);
    getchar();
    for (int i = 0; i < N; i++)
    {
        char a,b;
        scanf("%c %c",&a,&b);
        int an,bn;
        getchar();
        if (a=='-')
        {
            an=-1;
        }else
        {
            an=a-'0';
        }
        
        if (b=='-')
        {
            bn=-1;
        }else
        {
            bn=b-'0';
        }
        t[i].l = an;
        t[i].r = bn;
        if (an>=0)
        {
            r[an]++;
        }
        if (bn>=0)
        {
            r[bn]++;
        }
        
    }
    //2
    int root;
    for (int i = 0; i < N; i++)
    {
        if (r[i] == 0)
        {
            root = i;
            break;
        }
    }
    //3
    queue<int> q;
    q.push(root);
    vector<int> levelOrder;
    while (q.size() != 0)
    {
        int f = q.front();
        levelOrder.push_back(f);
        q.pop();
        if (t[f].r >= 0)
        {
            q.push(t[f].r);
        }
        if (t[f].l >= 0)
        {
            q.push(t[f].l);
        }
    }
    int levelOrdersize = levelOrder.size();
    for (int i = 0; i < levelOrdersize-1; i++)
    {
        printf("%d ",levelOrder[i]);
    }
    if (levelOrdersize)
    {
        printf("%d",levelOrder[levelOrdersize-1]);
    }
    printf("\n");
    //4
    stack<int> s;
    s.push(root);
    vector<int> reversedInorder;
    vector<int> visited(N);
    while (s.size()!=0)
    {
        int top=s.top();
        if (t[top].r >= 0 && visited[t[top].r] == 0)
        {
            s.push(t[top].r);
            continue;
        }
        reversedInorder.push_back(top);
        s.pop();
        visited[top]++;
        if (t[top].l >= 0 && visited[t[top].l] == 0)
        {
            s.push(t[top].l);
        }
    }
    int reversedInordersize = reversedInorder.size();
    for (int i = 0; i < reversedInordersize - 1; i++)
    {
        printf("%d ", reversedInorder[i]);
    }
    if (reversedInordersize)
    {
        printf("%d", reversedInorder[reversedInordersize - 1]);
    }
    return 0;
}

运行用时

结尾

看在我写了这么多注释的份上可以给我点个赞嘛,求求惹=]砰砰砰,给我加点写下去的油呀@.@
也欢迎关注我的CSDN账号呀,接下来两个月我应该会按这个格式更新所有的PTA甲级题目

                                        **开心code每一天**