PAT 1102 Invert a Binary Tree

Google: 90% of our engineers use the software you wrote (Homebrew), but you can’t invert a binary tree on a whiteboard so fuck off. ——————————-Max Howell @twitter

哈哈哈这句话实在是太有趣了。
既然是为了输出遍历序列的话，那就没有必要去重新构建一颗新的树。需要注意的是，由于scanf的%c格式是可以读入换行符的，因此需要在每行输入前把还是那个一行的换行符接受了。getchar是可以的，另外scanf(“%*c”)也是可以的。

#include <iostream>
#include <stdio.h>
#include <cstring>
#include <cstdio>
#include <map>
#include <math.h>
#include <string.h>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
using namespace std;

const int maxn = 11;
int n;
int num = 0;
int pre[maxn],in[maxn],post[maxn];
int flag[maxn] = {0};

int node[maxn][2];

void invert_level(int root)
{
    queue<int> Q;
    Q.push(root);

    while(!Q.empty())
    {
        int tmp;
        tmp = Q.front();
        Q.pop();
        printf("%d",tmp);
        num++;
        if(num < n)
            printf(" ");
        else
            printf("\n");

        if(node[tmp][1]!=-1)
            Q.push(node[tmp][1]);
        if(node[tmp][0]!=-1)
            Q.push(node[tmp][0]);
    }
}

void invert_in(int root)
{
    if(root == -1)
        return;
    invert_in(node[root][1]);
    num++;
    printf("%d",root);
    if(num<n)
        printf(" ");
    invert_in(node[root][0]);
}

int main()
{
    int i;
    char j,k;
    scanf("%d",&n);
    for(i=0;i<n;i++)
    {
        scanf("%*c%c %c", &j,&k);

        if(j!='-')
        {
            node[i][0]= j-'0';
            flag[j-'0'] = 1;
        }
        else
            node[i][0] = -1;

        if(k!='-')
        {
            node[i][1] = k-'0';
            flag[k-'0'] = 1;
        }
        else
            node[i][1] = -1;
    }
    int root;

    for(i=0;i<n;i++)
        if(flag[i]==0)
        {
            root = i;
            break;
        }

    invert_level(root);
    num = 0;
    invert_in(root);

    return 0;
}

然后是算法笔记的invert版本。核心就是用后序遍历去翻转二叉树，因为后序遍历会先访问左边的节点，再访问右边的节点。访问完了以后，再访问根节点。所以这样就可以在访问根节点的时候把左右儿子互换一下。

postOrder(Node[root].lchild);
postOrder(Node[root].rchild);
swap(Node[root].lchild, Node[root].rchild);

这是反转了以后我再去进行其他遍历的例子。总的代码如下：

#include <cstdio>
#include <queue>
#include <algorithm>
using namespace std;

const int maxn = 11;
struct node{
    int lchild,rchild;
}Node[maxn];

bool notRoot[maxn] = {false};
int num,n = 0;

void print(int id)
{
    printf("%d", id);
    num++;
    if(num<n)
        printf(" ");
    else
        printf("\n");
}

void inOrder(int root)
{
    if(root == -1)
        return;
    inOrder(Node[root].lchild);
    print(root);
    inOrder(Node[root].rchild);
}

void bfs(int root){
    queue<int> q;
    q.push(root);
    while(!q.empty())
    {
        int now = q.front();
        q.pop();
        print(now);
        if(Node[now].lchild!=-1)
            q.push(Node[now].lchild);
        if(Node[now].rchild!=-1)
            q.push(Node[now].rchild);
    }
}

void postOrder(int root)
{
    if(root == -1)
        return;
    postOrder(Node[root].lchild);
    postOrder(Node[root].rchild);
    swap(Node[root].lchild, Node[root].rchild);
}

int strToNum(char c)
{
    if(c == '-')
        return -1;
    else
    {
        notRoot[c - '0'] = true;
        return c - '0';
    }

}

int findRoot()
{
    for(int i = 0;i<n;i++)
    {
        if(notRoot[i] == false)
            return i;
    }
    return -1;
}

int main()
{
    char lchild,rchild;
    scanf("%d", &n);
    for(int i=0;i<n;i++)
    {
        scanf("%*c%c %c", &lchild, &rchild);
        Node[i].lchild = strToNum(lchild);
        Node[i].rchild = strToNum(rchild);
    }
    int root = findRoot();
    postOrder(root);
    bfs(root);
    num = 0;
    inOrder(root);
    return 0;
}