运用递归来解决树的问题

参考 : 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;
}
*/

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值