LeetCode ---- 二叉树剪枝

给定二叉树根结点 root ,此外树的每个结点的值要么是 0,要么是 1。

返回移除了所有不包含 1 的子树的原二叉树。

( 节点 X 的子树为 X 本身,以及所有 X 的后代。)

示例1:

输入: [1,null,0,0,1]
输出: [1,null,0,null,1]
这里写图片描述
解释:
只有红色节点满足条件“所有不包含 1 的子树”。
右图为返回的答案。

示例2:

输入: [1,0,1,0,0,0,1]
输出: [1,null,1,null,1]
这里写图片描述

示例3:

输入: [1,1,0,1,1,0,1,0]
输出: [1,1,0,1,1,null,1]
这里写图片描述

说明:

给定的二叉树最多有 100 个节点。
每个节点的值只会为 0 或 1 。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
/**
 * -------------------------------------------------<br>
 * Copyright (c) 2015-2017 AIA . All Rights Reserved.  <br>
 * -------------------------------------------------<br>
 * Project Name   : <code>LeetCode</code><br>
 *
 * @author: Tal.Yuan
 * @version: 2.0
 * Description:
 */
package com.tal.leetcode;

import org.junit.Assert;
import org.junit.Test;

import java.util.Stack;


public class PruneTree {

    @Test
    public void pruneTreeTest() {
        TreeNode test = new TreeNode(1);
        test.right = new TreeNode(0);
        test.right.left = new TreeNode(0);
        test.right.right = new TreeNode(1);

        TreeNode actual = pruneTree(test);

        Assert.assertEquals(1, actual.val);
        Assert.assertNull(actual.left);
        Assert.assertEquals(0, actual.right.val);
        Assert.assertNull(actual.right.left);
        Assert.assertEquals(1, actual.right.right.val);
    }

    public TreeNode pruneTree(TreeNode root) {
        TreeNode temp = root;
        Stack<TreeNode> stack = new Stack<>();
        while (root != null || stack.size() > 0) {
            while (root != null) {
                if (root.left != null && checkInvalidNode(root.left)) {
                    root.left = null;
                }

                if (root.right != null) {
                    stack.push(root);
                }

                root = root.left;
            }

            if (!stack.isEmpty()) {
                TreeNode treeNode = stack.pop();
                if (null != treeNode && checkInvalidNode(treeNode.right)) {
                    treeNode.right = null;
                }
                root = treeNode.right;
            }
        }

        return temp;
    }

    public boolean checkInvalidNode(TreeNode root) {
        Stack<TreeNode> stack = new Stack<>();
        while (root != null || !stack.isEmpty()) {
            while (root != null) {
                if (root.val == 1) {
                    return false;
                }
                stack.push(root);
                root = root.left;
            }

            if (!stack.isEmpty()) {
                TreeNode treeNode = stack.pop();
                if (null != treeNode && null != treeNode.right && treeNode.right.val == 1) {
                    return false;
                }
                root = treeNode.right;
            }
        }
        return true;
    }

}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
LeetCode-Editor是一种在线编码工具,它提供了一个用户友好的界面编写和运行代码。在使用LeetCode-Editor时,有时候会出现乱码的问题。 乱码的原因可能是由于编码格式不兼容或者编码错误导致的。在这种情况下,我们可以尝试以下几种解决方法: 1. 检查文件编码格式:首先,我们可以检查所编辑的文件的编码格式。通常来说,常用的编码格式有UTF-8和ASCII等。我们可以将编码格式更改为正确的格式。在LeetCode-Editor中,可以通过界面设置或编辑器设置来更改编码格式。 2. 使用正确的字符集:如果乱码是由于使用了不同的字符集导致的,我们可以尝试更改使用正确的字符集。常见的字符集如Unicode或者UTF-8等。在LeetCode-Editor中,可以在编辑器中选择正确的字符集。 3. 使用合适的编辑器:有时候,乱码问题可能与LeetCode-Editor自身相关。我们可以尝试使用其他编码工具,如Text Editor、Sublime Text或者IDE,看是否能够解决乱码问题。 4. 查找特殊字符:如果乱码问题只出现在某些特殊字符上,我们可以尝试找到并替换这些字符。通过仔细检查代码,我们可以找到导致乱码的特定字符,并进行修正或替换。 总之,解决LeetCode-Editor乱码问题的方法有很多。根据具体情况,我们可以尝试更改文件编码格式、使用正确的字符集、更换编辑器或者查找并替换特殊字符等方法来解决这个问题。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值