1. 题⽬链接:129.求根节点到叶节点数字之和
2. 题⽬描述:
3. 解法(dfs-前序遍历):
前序遍历按照根节点、左⼦树、右⼦树的顺序遍历⼆叉树的所有节点,通常⽤于⼦节点的状态依赖于 ⽗节点状态的题⽬。
算法思路:
在前序遍历的过程中,我们可以往左右⼦树传递信息,并且在回溯时得到左右⼦树的返回值。递归函 数可以帮我们完成两件事:
1. 将⽗节点的数字与当前节点的信息整合到⼀起,计算出当前节点的数字,然后传递到下⼀层进⾏递 归;
2. 当遇到叶⼦节点的时候,就不再向下传递信息,⽽是将整合的结果向上⼀直回溯到根节点。 在递归结束时,根节点需要返回的值也就被更新为了整棵树的数字和。
算法流程:
递归函数设计:int dfs(TreeNode* root,int num) 1. 返回值:当前⼦树计算的结果(数字和);
2. 参数num:递归过程中往下传递的信息(⽗节点的数字);
3. 函数作⽤:整合⽗节点的信息与当前节点的信息计算当前节点数字,并向下传递,在回溯时返回当 前⼦树(当前节点作为⼦树根节点)数字和。
递归函数流程:
1. 当遇到空节点的时候,说明这条路从根节点开始没有分⽀,返回0;
2. 结合⽗节点传下的信息以及当前节点的val,计算出当前节点数字sum;
3. 如果当前结点是叶⼦节点,直接返回整合后的结果sum;
4. 如果当前结点不是叶⼦节点,将sum传到左右⼦树中去,得到左右⼦树中节点路径的数字和,然 后相加后返回结果。
C++算法代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution
{
public:
//叶子结点返回最低端的值,非叶子结点返回左右最低端之和
int dfs(TreeNode* node,int temp)
{
temp=temp*10+node->val;
if(node->left==nullptr&&node->right==nullptr)
{
return temp;
}
int ret=0;
if(node->left)
{
ret+=dfs(node->left,temp);
}
if(node->right)
{
ret+=dfs(node->right,temp);
}
return ret;
}
int sumNumbers(TreeNode* root)
{
return dfs(root,0);
}
};
Java算法代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution
{
public int sumNumbers(TreeNode root)
{
return dfs(root, 0);
}
public int dfs(TreeNode root, int preSum)
{
preSum = preSum * 10 + root.val;
if (root.left == null && root.right == null)
return preSum;
int ret = 0;
if (root.left != null) ret += dfs(root.left, preSum);
if (root.right != null) ret += dfs(root.right, preSum);
return ret;
}
}