参考 : https://leetcode-cn.com/leetbook/read/data-structure-binary-tree/xefb4e/
对于树,我们可以通过 "自顶向下"(top_down) 和 "自底向上" (bottom_up)的递归来解决树的问题。
自顶向下(类似于前序遍历)
"自顶向下",意味着在每个递归层级,我们将访问该节点的值来计算一些值;
然后在递归调用时,将这些计算得到的值传递到子节点上。
比如,给定一个二叉树,寻找它的最大深度?
我们知道根的深度为 1。对于每个节点,如果我们知道该节点的深度,则自然知道它左右子节点的深度。
因此,在调用递归的时候,我们需要将节点的深度作为一个参数,一直传递下去,那么每个节点都知道自己所处的深度。
而递归终止条件就是叶子节点,当递归到叶子节点时,我们应该将该叶子节点的深度与树的最大深度做比较,并更新树的最大深度。
一般自顶向下的递归函数,是不需要返回值的,因为所需要的参数都是从上面传下来的。所以参数列表就需要考虑传哪些值给子节点。
同样需要注意的是,递归都是要有终止条件的,比如求二叉树最大深度,它的终止条件就是叶子节点。
int ans = 0; // 全局变量,用来记录树的最大深度
/*
对于递归函数 depth(TreeNode* node, int curDep) 的定义
当前 node 节点的深度是 curDep
递归终止条件 : 当前节点为叶子节点,则更新最大深度
*/
void depth(TreeNode* node, int curDep) { // 递归函数, curDep 表示当前节点 node 的深度
if (node) { // 防止树的根节点为空
if (node->left == nullptr && node->right == nullptr) { // 递归终止条件,若该节点是叶子节点,则更新树的最大深度
ans = ans > curDep ? ans : curDep;
return;
}
if (node->left) { // 若该节点有左孩子,则递归求解左子树的深度,注意,curDep + 1 传入下一层
depth(node->left, curDep + 1);
}
if (node->right) {
depth(node->right, curDep + 1);
}
}
}
int maxDepth(TreeNode* root) {
depth(root, 1); // root 节点当前的深度为 1
return ans;
}
自底向上(类似于后序遍历)
在每个递归层级上,我们首先对该节点的所有子节点递归的调用函数
然后根据其返回的值和该节点本身的值得到答案
继续讨论二叉树的最大深度,这次我们换自底向上的思考方式。
假设,我们知道某个节点左右子树的深度,那么该节点的深度是否可知? 当然可知,该节点的深度就是左右子树中最大的那个再加1即可。
那么我们便可以构造一个自底向上的递归函数,该函数接收一个节点,然后返回该节点的最大深度。
在函数里面,我们可以递归的获取该节点的左子树深度以及右子树深度,然后对比得出最大值,则当前节点的最大深度就是左右子树最大值加1。
递归函数的终止条件就是空节点,空节点的深度为 0。
一般自底向上的递归函数,都是需要返回值的。因为函数里面需要先获取子节点递归的返回值,然后根据这些返回值得出当前节点的返回值。
同样需要注意的是,我们要善于找到递归终止的条件,这有利于我们理解返回值的含义。
/*
递归函数 int maxDepth(TreeNode* root) 的定义 :
以 root 为根节点的最大深度
递归函数终止条件 :
当前节点为 nullptr,则深度为 0 , 返回 0。
*/
int maxDepth(TreeNode* root) {
if (root == nullptr) return 0;
int leftDep = maxDepth(root->left); // 当前节点左子树的最大深度
int rightDep = maxDepth(root->right); // 当前节点右子树的最大深度
int curDep = (leftDep > rightDep) ? (leftDep + 1) : (rightDep + 1); // 当前节点的最大深度 = 左右子树中最大深度 + 1
return curDep;
}
/*
// 简化版
int maxDepth(TreeNode* root) {
return root == nullptr ? 0 : max(maxDepth(root->left), maxDepth(root->right)) + 1;
}
*/