给定一棵二叉树的根 root,请你考虑它所有 从根到叶的路径:从根到任何叶的路径。(所谓一个叶子节点,就是一个没有子节点的节点)
假如通过节点 node 的每种可能的 “根-叶” 路径上值的总和全都小于给定的 limit,则该节点被称之为「不足节点」,需要被删除。
请你删除所有不足节点,并返回生成的二叉树的根。
示例 1:
输入:root = [1,2,3,4,-99,-99,7,8,9,-99,-99,12,13,-99,14], limit = 1
输出:[1,2,3,4,null,null,7,8,9,null,14]
示例 2:
输入:root = [5,4,8,11,null,17,4,7,1,null,null,5,3], limit = 22
输出:[5,4,8,11,null,17,4,7,null,null,null,5]
示例 3:
输入:root = [5,-6,-6], limit = 0
输出:[]
提示:
给定的树有 1 到 5000 个节点
-10^5 <= node.val <= 10^5
-10^9 <= limit <= 10^9
思路:对于每个节点,我们通过后序遍历找到从该节点为起点的所有路径中的最大值,然后我们重新从根节点出发按照先序遍历的顺序判断该点是否需要删除,在先序遍历的过程中我们可以记录走到该节点所经过的路径值的总和,并与当前节点已处理的向下走所能得到的最大路径值求和,判断是否小于limit即可。
class Solution {
Map<TreeNode, Integer> map;
public TreeNode sufficientSubset(TreeNode root, int limit) {
if (root == null)
return null;
map = new HashMap<>();
map.put(root, back_dfs(root));
if (map.get(root) < limit)
return null;
mid_dfs(null, root, 0, limit, 0);
return root;
}
private void mid_dfs(TreeNode last, TreeNode root, int sum, int limit, int id) {
if (root == null)
return;
if (sum + map.get(root) < limit) {
if (last == null) root = null;
else {
if (id == 0) last.left = null;
else last.right = null;
}
return;
}
mid_dfs(root, root.left, sum + root.val, limit, 0);
mid_dfs(root, root.right, sum + root.val, limit, 1);
}
private int back_dfs(TreeNode root) {
if (root == null)
return 0;
int mx = Integer.MIN_VALUE;
if (root.left != null)
mx = Math.max(mx, back_dfs(root.left));
if (root.right != null)
mx = Math.max(mx, back_dfs(root.right));
if (mx == Integer.MIN_VALUE)
mx = 0;
map.put(root, mx + root.val);
return root.val + mx;
}
}