网易 | 题解 | #树上摘樱桃#
解析
“樱桃”串 :有两个为叶子节点的孩子节点
实现
HashMap<Integer, int[]> map = new HashMap();//利用HashMap记录所有节点及其左右子节点
10 9
1 left 2
1 right 3
2 left 4
2 right 5
3 right 6
6 left 7
6 right 8
8 left 9
8 right 10
map结构:
{1:2,3}
{2:4,5}
{3:6,-1} // 空节点用-1表示
{6:7,8}
{8:9,10}
// 然后遍历判断即可
int res = 0;
Iterator<Map.Entry<Integer, int[]>> iterator = map.entrySet().iterator();
while (iterator.hasNext()) {
Map.Entry<Integer, int[]> next = iterator.next();
int[] value = next.getValue();
if (value[0] == -1 || value[1] == -1) {//孩子节点为空
continue;
}
if (!map.containsKey(value[0]) && !map.containsKey(value[1])) {//两个叶子节点
res++;
}
}
System.out.println(res);