非递归实现前序、中序、后序遍历

下面的三组遍历均需要使用到栈,此处给出栈的简单定义,使用数组保存数据。使用数的节点时,传递的都是点对应的数组索引。(是i不是a[i])

//定义栈及其方法
struct Stack
{
    int a[N]; //存入栈元素,N为二叉树的点个数
    int top; //栈顶指针
};
typedef struct Stack Stack;

void addStack(Stack *s, int data) //入栈
{
    s->top++;
    s->a[s->top] = data;
}

int delStack(Stack *s) //出栈
{
    return s->a[s->top--];
}

int getTop(Stack *s) //得到栈顶元素
{
    return s->a[s->top];
}

1、前序遍历。
a. 初始化栈,将栈顶指针置-1;
b. 初始化标记数组label为0;
c. 输出根节点,标记根节点,根节点入栈;
d. 栈不为空的循环判断;
f. 得到栈顶元素(不是出栈),先判断是否有左孩子而且未标记,满足则输出、标记、入栈左孩子。在判断是否有右孩子而且未标记,满足则输出、标记、入栈右孩子。不满足这两个条件,弹出该元素。
g. 释放栈空间。

void preOrder(int root)
{
    Stack *s = (Stack *)malloc(sizeof(Stack));
    s->top = -1;
    int label[N] = {0};
    printf("%d ", a[root]);
    label[root] = 1;
    addStack(s, root); //根入栈
    while(s->top != -1)
    {
        int tmp = getTop(s);
        if(2*tmp+1 < N && a[2*tmp+1] != -1 && label[2*tmp+1] == 0) //有左孩子
        {
            printf("%d ", a[2*tmp+1]);
            label[2*tmp+1] = 1;
            addStack(s, 2*tmp+1);
        }
        else if(2*tmp+2 < N && a[2*tmp+2] != -1 && label[2*tmp+2] == 0) //有右孩子
        {
             printf("%d ", a[2*tmp+2]);
             label[2*tmp+2] = 1;
             addStack(s, 2*tmp+2);
        }
        else
            delStack(s); //当其没有孩子,出栈
    }
    free(s);
}

2、中序遍历。
a. 初始化栈空间,栈顶指针,标记数组,根入栈;
b. 循环判断栈不为空;
c. 得到栈顶元素,如果它有左孩子,将左孩子标记,并入栈左孩子;如果它有右孩子,先输出该元素,弹出该元素,标记入栈右孩子;如果没有孩子,输出该元素,弹出该元素。
注意:在每次输出该元素之后,一定要弹出该元素,否则会产生遍历多次的情况。

void midOrder(int root)
{
    Stack *s = (Stack *)malloc(sizeof(Stack));
    s->top = -1;
    int label[N] = {0};
    label[root] = 1;
    addStack(s, root);
    while(s->top != -1)
    {
        int tmp = getTop(s);
        if(2*tmp+1 < N && a[2*tmp+1] != -1 && label[2*tmp+1] == 0)
        {
            addStack(s, 2*tmp+1);
            label[2*tmp+1] = 1;
        }
        else if(2*tmp+2 < N && a[2*tmp+2] != -1 && label[2*tmp+2] == 0)
        {
            printf("%d ", a[tmp]);
            delStack(s);
            addStack(s, 2*tmp+2);
            label[2*tmp+2] = 1;
        }
        else
        {
            printf("%d ", a[tmp]);
            delStack(s);
        }
    }
    free(s);
}

3、后序遍历。
a. 初始化栈空间,栈顶指针,标记数组,根入栈;
b. 循环判断栈不为空;
c. 得到栈顶元素,如果它有左孩子,标记入栈左孩子。如果它有右孩子,标记压栈右孩子。如果都不满足,输出该元素,弹出该元素。

void bacOrder(int root)
{
    Stack *s = (Stack *)malloc(sizeof(Stack));
    s->top = -1;
    int label[N] = {0};
    label[root] = 1;
    addStack(s, root);
    while(s->top != -1)
    {
        int tmp = getTop(s);
        if(2*tmp+1 < N && a[2*tmp+1] != -1 && label[2*tmp+1] == 0)
        {
            addStack(s, 2*tmp+1);
            label[2*tmp+1] = 1;
        }
        else if(2*tmp+2 < N && a[2*tmp+2] != -1 && label[2*tmp+2] == 0)
        {
            addStack(s, 2*tmp+2);
            label[2*tmp+2] = 1;
        }
        else
        {
            printf("%d ", a[tmp]);
            delStack(s);
        }
    }
    free(s);
}

总结:三种遍历方法相似,唯一需要注意的地方就是输出元素和弹出元素的位置区别。

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,下面是递归非递归实现二叉树前序中序后序遍历算法代码。 ```python # 定义二叉树结构体 class TreeNode: def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right # 递归实现前序遍历 def preorder_recursive(root): if not root: return [] return [root.val] + preorder_recursive(root.left) + preorder_recursive(root.right) # 非递归实现前序遍历 def preorder_non_recursive(root): if not root: return [] res, stack = [], [root] while stack: node = stack.pop() res.append(node.val) if node.right: stack.append(node.right) if node.left: stack.append(node.left) return res # 递归实现中序遍历 def inorder_recursive(root): if not root: return [] return inorder_recursive(root.left) + [root.val] + inorder_recursive(root.right) # 非递归实现中序遍历 def inorder_non_recursive(root): if not root: return [] res, stack, node = [], [], root while stack or node: while node: stack.append(node) node = node.left node = stack.pop() res.append(node.val) node = node.right return res # 递归实现后序遍历 def postorder_recursive(root): if not root: return [] return postorder_recursive(root.left) + postorder_recursive(root.right) + [root.val] # 非递归实现后序遍历 def postorder_non_recursive(root): if not root: return [] res, stack = [], [root] while stack: node = stack.pop() if node.left: stack.append(node.left) if node.right: stack.append(node.right) res.append(node.val) return res[::-1] ``` 这里给出了二叉树前序中序后序遍历递归实现非递归实现,其中非递归实现使用了栈来实现

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值