计算和的过程就是dfs。
dfs的过程中传输到当前节点的和,判断出传递到叶节点的时候,将当前和加入到sum中。
需要用map存储每个位置对应的值。此处就涉及到堆中的一个概念:当前节点与它的左右子节点之间的索引的计算关系。以及当前节点的索引与它的深度和所在深度的位置关系。
id = 2^(depth-1) + p -1
public class test {
Map<Integer, Integer> map = new HashMap<>();
int sum = 0;
public int pathSum(int[] num){
if(num.length == 0){
return 0;
}
for(int i=0; i<num.length; i++){
int pos = (1<<(num[i]/100-1)) + num[i]/10%10 -1;
map.put(pos, num[i]%10);
}
dfs(1,0);
return sum;
}
void dfs(int p, int s){
if(!map.containsKey(p)){
return;
}
s += map.get(p);
boolean l = map.containsKey(2*p);
boolean r = map.containsKey(2*p+1);
if(!l && !r){
sum += s;
return;
}
dfs(2*p,s);
dfs(2*p+1,s);
}
}