我们可以看到从根节点到当前节点这条路径的值就是父节点的值*2加上当前节点的值。我们定义一个全局的变量res,他就是所有从根节点到叶子节点表示数字的和。
我们可以通过前序遍历来解这道题,当遇到叶子节点的时候就把从根节点到当前叶子节点表示的数字加到res中。直接把二叉树的前序遍历方式修改一下即可。
//最终返回的数字
int res = 0;
public int sumRootToLeaf(TreeNode root) {
dfs(root, 0);
return res;
}
//parentPathSum表示从根节点到当前父节点这条路径表示的数字
public void dfs(TreeNode root, int parentPathSum) {
//如果节点为空,直接返回
if (root == null)
return;
//父节点的值\*2,在加上当前节点的值就是从根节点到
//当前节点这条路径表示的数字
int sum = parentPathSum \* 2 + root.val;
//如果到叶子节点,说明找到了一个从根节点到叶子
//节点的完整路径,把这条路径的值加到res中
if (root.left == null && root.right == null) {
res += sum;
return;
}
//如果没到叶子节点就继续遍历当前节点的左子节点和右子节点
dfs(root.left, sum);
dfs(root.right, sum);
}
时间复杂度:O(N),N是节点的个数,所有节点都要访问一遍
空间复杂度:O(H),H是树的最大高度,也是栈的深度
BFS解决
除了DFS,我们还可以使用BFS来解决,DFS就是深度优先搜索,BFS就是广度优先搜索,具体也可以看下373,数据结构-6,树。BFS就是一层一层的访问。这里需要使用两个队列:
- 一个存放节点
- 一个存放从根节点到当前节点这条路径表示的数字
如果访问到叶子节点的时候就把表示的数字加入到res中,最后返回res即可,我们来看下代码。
public int sumRootToLeaf(TreeNode root) {
int res = 0;//结果值
//两个队列,一个存放节点,一个存放从根节点到当前
//节点的父节点这条路径所表示的数字
Queue<TreeNode> queueNode = new LinkedList<>();
Queue<Integer> queueParentSum = new LinkedList<>();
queueNode.add(root);
queueParentSum.add(0);
while (!queueNode.isEmpty()) {
TreeNode cur = queueNode.poll();
int parentSum = queueParentSum.poll();
//计算从根节点到当前节点这条路径表示的数字
int sum = parentSum \* 2 + cur.val;
//如果当前节点是叶子节点,就把sum加到res中
if (cur.left == null && cur.right == null) {
res += sum;
continue;
}
//如果左子节点不为空,就把他和他对应的值分别加入
//到对应的队列中
if (cur.left != null) {
queueNode.add(cur.left);
queueParentSum.add(sum);
}
//右子节点同上
if (cur.right != null) {
### 给大家的福利
**零基础入门**
对于从来没有接触过网络安全的同学,我们帮你准备了详细的学习成长路线图。可以说是最科学最系统的学习路线,大家跟着这个大的方向学习准没问题。
![](https://img-blog.csdnimg.cn/img_convert/95608e9062782d28f4f04f821405d99a.png)
同时每个成长路线对应的板块都有配套的视频提供:
![在这里插入图片描述](https://img-blog.csdnimg.cn/direct/a91b9e8100834e9291cfcf1695d8cd42.png#pic_center)
因篇幅有限,仅展示部分资料