2021.04.30DFS求好叶子节点对数量
(题目来源:https://leetcode-cn.com/problems/number-of-good-leaf-nodes-pairs/)
题目描述
给你二叉树的根节点 root 和一个整数 distance 。
如果二叉树中两个 叶 节点之间的 最短路径长度 小于或者等于 distance ,那它们就可以构成一组 好叶子节点对 。
返回树中 好叶子节点对的数量 。
思路
- 在二叉树的题目中,首先考虑dfs。
- dfs应该返回的信息:该节点下面叶子节点到当前位置的距离 和 个数。
- 采用数组int[] cnt =new int[n+1],cnt[i]表示叶子节点到当前位置距离为i的叶节点个数。
代码
//自底向上的递归,用count数组统计和当前节点的距离为 i 的叶子节点数目,
//多于distance的不用考虑
int[] dfs(TreeNode cur, int distance){
int[] cnt = new int[distance+1];
if(cur == null) return cnt;
if(cur.left == null && cur.right == null) {//其本身就是叶子节点
//所以其父亲节点下 距其距离为1的节点个数有1个。
cnt[1] = 1;
return cnt;
}
int[] left = dfs(cur.left,distance);
int[] right = dfs(cur.right,distance);
for(int i = 1; i < distance; i++) {
for(int j = 1; j <= distance-i; j++) {
//求对数
ans+=left[i]*right[j];
}
}
//合并数组,求得当前节点的返回结果
for(int i = 2; i <= distance; i++) {
cnt[i] = left[i-1]+right[i-1];
}
return cnt;
}
2021.6.1 再看此题
本题要求找到一棵树的叶子节点之间的关系,所以应该从叶子节点向根节点遍历。每次遍历都返回之前遍历状态。
本题中,返回的状态用int数组记录,为当前节点下的叶子节点的好叶子节点个数。如下图:
遍历A、B子树后,返回A、B子树的遍历状态和叶子节点的信息,此时开始遍历2号节点。遍历2号节点时,由于得到两树的状态信息,选择不再比较A,B树内部细节,而是比较A,B之间的叶子节点,看是否满足好叶子节点的关系。
遍历完2,3号节点后,开始遍历1号节点,此时也是如此。不再考虑颜色相同子树内部的信息(如果题目要求不同,也是可以考虑的),而是考虑蓝绿子树之间的信息。