二叉树根节点到叶子节点的所有路径和
题目: 给定一个仅包含数字 0?9 的二叉树,每一条从根节点到叶子节点的路径都可以用一个数字表示。 例如根节点到叶子节点的一条路径是1→2→3,那么这条路径就用 123 来代替。 找出根节点到叶子节点的所有路径表示的数字之和 例如: 这颗二叉树一共有两条路径, 根节点到叶子节点的路径 1→2 用数字 12 代替 根节点到叶子节点的路径 1→3 用数字 13 代替 所以答案为 12+13=25
思路
这道题很简单,只是我一开始思路想错了,并且在线上环境,不可以用IDEA,导致我算平方和要自己写。因为一开始思路错了,最后换思路的时候时间太紧了,我的路径也记错了(漏了第一个节点),所以最后没有AC。 就是用一个全局变量记录答案,然后有一个数组记录所有的路径,到了叶子节点就开始计算路径和。 上代码:
public int max(ThreeNode node){
List<Integer> nodes = new ArrayList<>();
nodes.add(node.val);
dfs(node,nodes);
return count1;
}
private int count1 = 0;
public void dfs(ThreeNode node, List<Integer> nodes) {
if(node.left == null && node.right == null) {
int temp = 0;
for (int i = 0; i < nodes.size(); i++){
int temp1 = 1;
for (int j = 0; j < nodes.size() - i - 1; j++){
temp1 *= 10;
}
temp += nodes.get(i) * temp1;
}
count1 += temp;
return;
}
if(node.left != null) {
nodes.add(node.left.val);
dfs(node.left,nodes);
nodes.remove(nodes.size() - 1);
}
if(node.right != null) {
nodes.add(node.right.val);
dfs(node.right,nodes);
nodes.remove(nodes.size() - 1);
}
}
做题还是要细心啊,思路理清楚再写,特别是线上的环境,只给十五分钟,很容易就把时间给浪费掉了。
|