代码随想录算法训练营第十四天 | 144. 二叉树的前序遍历，145.二叉树的后序遍历，94.二叉树的中序遍历 [二叉树篇]

二叉树的递归遍历

前言

二叉树的理论基础知识，可以看代码随想录中的《二叉树理论基础》。

二叉树遍历方式

二叉树主要有两种遍历方式：

• 深度优先遍历：先往深走，遇到叶子节点再往回走
• 广度优先遍历：一层一层地遍历

进一步可以分为

• 深度优先遍历
  • 前序遍历（中-左-右），主流方法：递归法、迭代法
  • 中序遍历（左-中-右），主流方法：递归法、迭代法
  • 后序遍历（左-右-中），主流方法：递归法、迭代法
• 广度优先遍历
  • 层次遍历，主流方法：迭代法

盗图说明一下各种遍历次序

递归法

视频讲解：每次写递归都要靠直觉？ 这次带你学透二叉树的递归遍历！

递归三要素

• 确定递归函数的参数和返回值
• 确定终止条件
• 确定单层递归的逻辑

进一步讲解下

1. 确定递归函数的参数和返回值
   需要三个参数：需要传入树节点的值，用来保存遍历节点数组的大小，以及遍历节点的数组。

int* traversal(struct TreeNode* node, int *arr, int *cnt)

2. 确定终止条件，当要遍历的节点为空时，本次遍历就要结束，需要返回。

	if (node == NULL) {
		return;
	}

3. 确定单层递归的逻辑，二叉树的遍历有三种：前序、中序、后序
   比如对于前序遍历，代码如下

	arr[*cnt++] = node->val;  // 中
	traversal(node->left, arr, cnt); // 左
	traversal(node->right, arr, cnt) // 右

对于递归遍历，中序遍历、后序遍历只需要将上面这三句代码换一下顺序。

迭代法

视频讲解：写出二叉树的非递归遍历很难么？（前序和后序）
视频讲解：写出二叉树的非递归遍历很难么？（中序）

大体思路是通过栈结构实现二叉树的前序、中序、后序遍历。

前序遍历

前序遍历的顺序：中-左-右
先将父节点pop进栈，然后再push出来，将节点放入到数组中。
接着判断父节点有没有右孩子，如果有，将右孩子也pop进栈，然后判断有父节点有没有左孩子，如果有，将左孩子也pop进栈。

待补充…

LeetCode 144. 二叉树的前序遍历

题目链接：144. 二叉树的前序遍历

题目描述

给你二叉树的根节点 root ，返回它节点值的 前序 遍历。

示例1

输入：root = [1,null,2,3]
输出：[1,2,3]

示例2

输入：root = []
输出：[]

示例3

输入：root = [1]
输出：[1]

提示

• 树中节点数目在范围 [0, 100] 内
• -100 <= Node.val <= 100

思路

二叉树的基本操作，可以使用递归方法和迭代方法等，具体可以看上面的讲解或者视频。

递归参考代码

void traversal(struct TreeNode* node, int *arr, int *cnt)
{
    if (node == NULL) {
        return;
    }

    arr[(*cnt)++] = node->val;  // 中
    traversal(node->left, arr, cnt); // 左
    traversal(node->right, arr, cnt); // 右
}

int* preorderTraversal(struct TreeNode* root, int* returnSize)
{
    int *arr = (int*)malloc(100*sizeof(int));
    int cnt = 0;

    traversal(root, arr, &cnt);
    *returnSize = cnt;

    return arr;
}

迭代参考代码

待实现

---

LeetCode 145.二叉树的后序遍历

题目链接：145.二叉树的后序遍历

题目描述

给你一棵二叉树的根节点 root ，返回其节点值的 后序遍历 。

示例1

输入：root = [1,null,2,3]
输出：[3,2,1]

示例2

输入：root = []
输出：[]

示例3

输入：root = [1]
输出：[1]

提示

• 树中节点数目在范围 [0, 100] 内
• -100 <= Node.val <= 100

思路

可以使用递归方法和迭代方法等，具体可以看上面的讲解或者视频。

递归参考代码

void traversal(struct TreeNode* node, int *arr, int *cnt)
{
    if (node == NULL) {
        return;
    }

    traversal(node->left, arr, cnt); // 左
    traversal(node->right, arr, cnt); // 右
    arr[(*cnt)++] = node->val;  // 中
}

int* postorderTraversal(struct TreeNode* root, int* returnSize) {
    int *arr = (int*)malloc(100*sizeof(int));
    int cnt = 0;

    traversal(root, arr, &cnt);
    *returnSize = cnt;

    return arr;
}

迭代参考代码

待实现

---

LeetCode 94. 二叉树的中序遍历

题目链接：94. 二叉树的中序遍历

题目描述

给定一个二叉树的根节点 root ，返回 它的 中序 遍历 。

示例1

输入：root = [1,null,2,3]
输出：[1,3,2]

示例2

输入：root = []
输出：[]

示例3

输入：root = [1]
输出：[1]

提示

• 树中节点数目在范围 [0, 100] 内
• -100 <= Node.val <= 100

思路

可以使用递归方法和迭代方法等，具体可以看上面的讲解或者视频。

递归参考代码

void traversal(struct TreeNode* node, int *arr, int *cnt)
{
    if (node == NULL) {
        return;
    }

    traversal(node->left, arr, cnt); // 左
    arr[(*cnt)++] = node->val;  // 中
    traversal(node->right, arr, cnt); // 右    
}

int* inorderTraversal(struct TreeNode* root, int* returnSize) {
    int *arr = (int*)malloc(100*sizeof(int));
    int cnt = 0;

    traversal(root, arr, &cnt);
    *returnSize = cnt;

    return arr;
}

迭代参考代码

待实现