非递归后序遍历二叉树

---

title: 非递归后序遍历二叉树
date: 2020-12-25 20:06:53
tags: 后序遍历
categories: 数据结构

---

使用栈实现后序遍历二叉树

写了很久很久很久，可能是自己太长时间没写这样的代码了吧，简直把自己写崩溃了，不过还好，终于算是写出来了。有的是写了才发现存储结构开的不对， 有的是平时没有遇到的比较神奇的bug 也不算是神奇吧，就是平时没有遇见过的问题，也算是有收获了。爱瞎搞，爱作，可不就遇到很多阴间问题呗~ hhhhhh

里边附带了一个自己瞎搞想出来的偷懒生成满二叉树的办法，可以用来当测试的背景

写一下自己的收获吧

1. 枚举体的默认值不是 值为0的那个，需要手动初始化
2. 树要带头指针
3. 栈中存的应该是树中结点的地址，而不是每个节点，要是把栈里的元素给设置成树的节点的话，好家伙，也太浪费内存了，并且，还不好操作，需要来回转换

#include <stdio.h>
#include <stdlib.h>
#define MAXSIZE 100

typedef enum
{
	unvisit, visit
}VisitTag;
typedef struct treeNode
{
    int data;
    struct treeNode *left, *right;
	VisitTag tag;
}TreeNode, *Tree;

// 栈里存的应该是每个结点的地址
typedef struct stack 
{
	Tree arr[MAXSIZE];
	int top;
} Stack;

Tree initTree(int);
void initStack(Stack *);
void postOrder(Tree, Stack *);
void inStack(Stack *, Tree);
Tree outStack(Stack *);

int main(void)
{
	Tree t = initTree(15);
	Stack stack;
	initStack(&stack);
	postOrder(t, &stack);
}

// 后序遍历
void postOrder(Tree t, Stack *s)
{
	Tree temp = t;
	// 当指针指向空且栈中没有元素时, 终止循环
	while(temp != NULL || s->top != 0)
	{
		while (temp != NULL)
		{
			inStack(s, temp);
			temp = temp->left;
		}

		Tree p = s->arr[(s->top) - 1];
		if (p->tag == unvisit)
		{  // 第一次被访问
			p->tag = visit;
			temp = p->right;
		}
		else
		{ // 第二次被访问, 栈顶元素出栈, 打印栈顶元素
			outStack(s);
			printf("%d\n", p->data);
		}
	}
}

// 初始化栈
void initStack(Stack *p)
{
	p->top = 0;
}

// 入栈
void inStack(Stack *s, Tree p)
{
	if (s->top >= MAXSIZE)
	{
		printf("栈已满, 入栈失败\n");
		return;
	}
	s->arr[s->top] = p;
	s->top++;
}

// 出栈
Tree outStack(Stack *s)
{
	if (s->top == 0)
	{
		printf("栈为空, 出栈失败\n");
		return NULL;
	}
	s->top--;
	return s->arr[s->top];
}

// 传进来的数必须满足 =  2 ^ n - 1 
Tree initTree(int nodeCounts)
{

    // 这个操作仅仅是让p指向申请的那块空间的首地址，p不是树的头指针
    Tree p = (Tree)malloc(sizeof(TreeNode) * (nodeCounts + 1));
    for (int i = 1; i <= nodeCounts; i++)
    {
        p[i].data = i;
        p[i].tag = unvisit;
    }
    for (int i = 1; i <= nodeCounts / 2; i++)
    {
        p[i].left = p + 2 * i;
        p[i].right = p + 2 * i + 1;
    }
    for (int i = nodeCounts / 2 + 1; i <= nodeCounts; i++)
    {
        p[i].left = NULL;
        p[i].right = NULL;
    }
    return p + 1;
}