二叉树遍历之迭代遍历

最新推荐文章于 2024-10-25 11:38:23 发布

原创最新推荐文章于 2024-10-25 11:38:23 发布 · 589 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构 #树

基本数据结构与算法（C语言版本）专栏收录该内容

21 篇文章

订阅专栏

前文我们简述了二叉树遍历之递归遍历

其实用到的是递归中自己的栈来完成的，那么我们要是用迭代的话也是要通过自己的栈来完成

前序遍历：根左右

我们在递归遍历时通过前序遍历的原理图示展示了整个过程：

根节点第一次出栈就可以打印了，然后再来分别出栈左右子节点

void preorderIteration(struct TreeNode* root, MyStack *stk)
{
	if(root == NULL)
		return;
		
	myStackPush(stk, root);
	
	while( ! myStackEmpty(stk) )
	{
		root = myStackTop(stk);
		myStackPop(stk);
		printf("val=%d\n", root->val);
		
		if(root->right)
			myStackPush(stk, root->right);
		
		if(root->left)
			myStackPush(stk, root->left);	
	}
}

中序遍历：左根右

我们遇见左节点时还要看有没有左子节点，一直要到最左边的叶子为止

所以根节点先入栈，然后持续对左子节点压栈，直到左子节点为NULL，这个时候就可以出栈左子节点，打印其值

左边节点值打印完成，如果其没有右子节点，那么再出栈的话就是其父节点（根）

如果有右子节点，那么该左节点视为子树的根节点，对右子树进行重复逻辑

void inorderIteration(struct TreeNode* root, MyStack *stk)
{
	if(root == NULL)
		return;
	
	while(! myStackEmpty(stk) || root != NULL)
	{
		if(root != NULL)
		{
			myStackPush(stk, root);
			root = root->left;
		}
		else
		{
			root = myStackTop(stk);
			myStackPop(stk);
			printf("val=%d\n", root->val);
			
			root = root->right;
		}
	
	}
}

1. 依次入栈根，左子节点，直到没有左子节点

2. 出栈，节点 4 没有右子节点，再出栈就是其父节点 2，同样节点 2也没有右子节点，再出栈就是节点 1

3. 节点 1 有右子节点，那么就开始处理右子树，整个过程逻辑一样

后序遍历：左，右，根

与前面两种方式不一样，我们在处理完左树后，根节点还不能打印

除非根节点没有右子树，或者右子树已经被打印了，才能打印根节点

所以根节点会进行二次入栈，当第二次出栈的时候通过比较临时节点变量（保存压栈时的根节点）与当前根节点是否一样，一样表示已经处理完，这个时候就能够打印当前根节点

void postorderIteration(struct TreeNode* root, MyStack *stk)
{
	if(root == NULL)
		return;
		
	struct TreeNode* prev = NULL;
	
	while(! myStackEmpty(stk) || root != NULL)
	{
		while(root != NULL)
		{
				myStackPush(stk, root);
				root = root->left;
		}
		
		root = myStackTop(stk);
		myStackPop(stk);
		
		if(root->right == NULL || root->right == prev)
		{
			printf("val=%d\n", root->val);
			prev = root;
			root = NULL;
		}
		else
		{
			myStackPush(stk, root);
			root = root->right;
		}
	}	
	
}

1. 根节点开始压栈，完成左子节点的压栈