三、删点成林(Weekly Contest 144)

给出二叉树的根节点 root,树上每个节点都有一个不同的值。

如果节点值在 to_delete 中出现,我们就把该节点从树上删去,最后得到一个森林(一些不相交的树构成的集合)。

返回森林中的每棵树。你可以按任意顺序组织答案。
示例:在这里插入图片描述

使用递归,转成list,判断当前的点是否应该去掉,如果可以返回null

class Solution {
  public List<TreeNode> delNodes(TreeNode root, int[] to_delete) {
        List<TreeNode> list = new ArrayList<TreeNode>();
        List<Integer> tem = new ArrayList<>();
        for (int integer : to_delete) {
			tem.add(integer);
		}
        if(root == null){
        	return list;
        }else {
        	root = dfs(root,tem, list);
        	if(root != null){
        		list.add(root);
        	}
		}
        return list;
    }
//	
	public TreeNode dfs(TreeNode root,List <Integer> tem,List<TreeNode> list){
		if(root == null){
			return null;
		}
		TreeNode left = dfs(root.left, tem, list);
		TreeNode right = dfs(root.right, tem, list);
		if(tem.contains(root.val)){
			if(left != null){
				list.add(left);
			}
			if(right != null){
				list.add(right);
			}
            return null;
		}
		root.left = left;
		root.right = right;
		return root;
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值