找到给定二叉树的对角线和


Question

给定一个二叉树,计算每个具有负斜率()的对角线的所有节点的和。 假设节点的左,右子与父节点成45度角。
考虑以下具有三个对角线的二叉树。 对角线的总和是10,15和11。

    1
   / 
  2   3
 /   / 
4   5   6
   / 
  7   8

对角线和分别为(7 + 4) = 11,(2 + 5 + 8) = 15,(1 + 3 + 6) =10
输出:11 15 10



Code

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
import java.util.Map;
import java.util.TreeMap;
class Node {
int data;
Node left, right;

Node(int item) {
data = item;
left = right = null;

}
}

public class BinaryTree {
Node root;

public void print(Node node, int dist, Map<Integer, Integer> map) {
if (node == null)
return;
if (!map.containsKey(dist)){
map.put(dist, node.data);
}else {
int sum = map.get(dist) + node.data;
map.put(dist, sum);
}

print(node.left, dist - 1, map);
print(node.right, dist, map);
}

public static void main(String[] args) {

BinaryTree tree = new BinaryTree();
Map<Integer, Integer> map = new TreeMap<>();// TreeMap保证添加进去的顺序
// Let us create trees shown in above diagram
tree.root = new Node(1);
tree.root.left = new Node(2);
tree.root.right = new Node(3);
tree.root.left.left = new Node(4);
tree.root.right.left = new Node(5);
tree.root.right.right = new Node(6);
tree.root.right.left.left = new Node(7);
tree.root.right.left.right = new Node(8);
tree.print(tree.root, 0, map);
for (Integer value : map.values()) {
System.out.print( value+ " ");
}


}
}