PAT Tree Traversals Again(根据中序遍历栈操作建树+后序遍历)

大体思路:

1.遇到push操作,就添加子节点,用一个栈保存父亲节点,设置flag标识左孩子还是右孩子;遇到pop操作,就取父亲节点;

2.用结构体数组实现二叉树,d[i]表示第i个节点

3.后序遍历递归实现(左右根)

#include <bits/stdc++.h>
using namespace std;
//根据中序遍历栈操作建树+后序遍历

int n, root, fa, node;
int flag, k=1;//flag=0标记下一个节点为左孩子节点,flag=1标记下一个节点为右孩子节点
string s, s1="Push", s2="Pop";
struct node
{
    int left;
    int right;
}d[100];
stack <int> myStack;

void postTraverse1(int root)//后序遍历(递归实现)
{
    if(root){
        postTraverse(d[root].left);
        postTraverse(d[root].right);
        printf("%d%c",root," \n"[k++==n]);
    }
}

int main()
{
    cin>>n;//读入共有多少节点
    while(cin>>s){
        if(s==s1) {//如果栈操作为push,增添子节点
            cin>>node;
            if(!root) root=node;//让第一个节点成为根 
            else{
                if(!flag) d[fa].left=node;
                else d[fa].right=node;
            }
            fa=node;flag=0;myStack.push(fa);
        }
        else if(s==s2) {//如果栈操作为pop,表示遍历到头
            flag=1;//下一个要添加的节点为右孩子节点
            fa=myStack.top();myStack.pop();//此节点成为父亲节点
        }
    }
    postTraverse(root);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/ChenyangXu/p/10731462.html