挑战刷leetcode第17天（二叉树-路径总和）

lqstyle
148 阅读4分钟

哪吒2之二叉树路径总和：从代码到思考，坚持的意义

大家好，我是哪吒2中的小哪吒，今天我要带大家进入一个神奇的世界——二叉树路径总和！别担心，我不会用乾坤圈和混天绫来解决这个问题，而是用Java和C++代码来搞定它。准备好了吗？让我们开始这段充满幽默与思考的旅程吧！

1. 背景故事：哪吒的二叉树冒险

话说有一天，哪吒在修炼时遇到了一个神秘的二叉树。这棵树的每个节点都有一个数值，哪吒的任务是找到所有从根节点到叶子节点的路径，使得路径上的节点值之和等于一个特定的目标值。听起来很简单？别急，这棵树可是有魔法的，路径可以不是从根节点开始，也可以不是到叶子节点结束。哪吒挠了挠头，决定用代码来解决这个问题。

2. Java版：哪吒的代码之旅

首先，我们来看看哪吒用Java写的代码：

public int pathSum(TreeNode root, long targetSum) {
    if (root == null) {
        return 0;
    }

    int ret = rootSum(root, targetSum);
    ret += pathSum(root.left, targetSum);
    ret += pathSum(root.right, targetSum);
    return ret;
}

public int rootSum(TreeNode root, long targetSum) {
    int ret = 0;

    if (root == null) {
        return 0;
    }
    int val = root.val;
    if (val == targetSum) {
        ret++;
    }

    ret += rootSum(root.left, targetSum - val);
    ret += rootSum(root.right, targetSum - val);
    return ret;
}

2.1 代码解析

  • pathSum方法：这是主方法，负责遍历整棵树。如果当前节点为空，返回0。否则，计算以当前节点为起点的路径总和，然后递归计算左子树和右子树的路径总和。
  • rootSum方法：这是辅助方法，负责计算以当前节点为起点的路径总和。如果当前节点为空，返回0。否则，检查当前节点的值是否等于目标值，如果是，计数器加1。然后递归计算左子树和右子树的路径总和，目标值减去当前节点的值。

2.2 思考时刻

  • 递归的魅力：哪吒发现，递归是解决二叉树问题的利器。通过递归，我们可以轻松地遍历整棵树，并处理每个节点。
  • 路径的多样性：路径不一定从根节点开始，也不一定到叶子节点结束。这增加了问题的复杂性，但也让哪吒更加深入地理解了二叉树的结构。

3. C++版：哪吒的代码再战

接下来，我们来看看哪吒用C++写的代码：

int pathSum(TreeNode* root, int targetSum) {
    if (root == nullptr) {
        return 0;
    }

    int ret = rootSum(root, targetSum);
    ret += pathSum(root->left, targetSum);
    ret += pathSum(root->right, targetSum);
    return ret;
}

int rootSum(TreeNode* root, long targetSum) {
    int ret = 0;

    if (root == nullptr) {
        return 0;
    }
    int val = root->val;
    if (val == targetSum) {
        ret++;
    }

    ret += rootSum(root->left, targetSum - val);
    ret += rootSum(root->right, targetSum - val);
    return ret;
}

3.1 代码解析

  • pathSum方法：与Java版类似，这是主方法，负责遍历整棵树。如果当前节点为空，返回0。否则，计算以当前节点为起点的路径总和，然后递归计算左子树和右子树的路径总和。
  • rootSum方法：与Java版类似，这是辅助方法，负责计算以当前节点为起点的路径总和。如果当前节点为空，返回0。否则，检查当前节点的值是否等于目标值，如果是，计数器加1。然后递归计算左子树和右子树的路径总和，目标值减去当前节点的值。

3.2 思考时刻

  • 语言的选择：哪吒发现，无论是Java还是C++，解决问题的思路是相同的。语言只是工具，关键在于算法的设计和理解。
  • 坚持的意义：哪吒在解决这个问题的过程中，遇到了很多困难，但他没有放弃。通过不断的学习和实践，他最终掌握了二叉树的路径总和问题。这告诉我们，坚持是成功的关键。

4. 总结：哪吒的修炼心得

通过这次二叉树路径总和的冒险，哪吒不仅学会了如何用Java和C++解决这个问题，还深刻体会到了坚持的意义。无论遇到多大的困难，只要我们不放弃，终会找到解决问题的方法。

希望这篇文章能让你在轻松愉快的氛围中学到知识，也希望你能像哪吒一样，在技术的道路上坚持不懈，勇往直前！

思考题：如果你遇到一个更复杂的二叉树问题，你会如何设计算法来解决它？欢迎在评论区分享你的想法！

作者坚持的意义：作为一名技术博客创作者，我始终坚持用通俗易懂的语言和幽默的风格来讲解复杂的技术问题。希望通过我的文章，能让更多的读者爱上编程，享受技术的乐趣。如果你喜欢这篇文章，请点赞、分享，并关注我的博客，更多精彩内容等你来发现！

avatar
文章
阅读
粉丝