修剪叶子

[编程题]修剪叶子

热度指数：3261 时间限制：C/C++ 1秒，其他语言2秒空间限制：C/C++ 256M，其他语言512M
算法知识视频讲解

有一棵有 $\mathit n$ 个节点的二叉树，其根节点为 $\mathit root$ 。修剪规则如下:

1.修剪掉当前二叉树的叶子节点，但是不能直接删除叶子节点

2.只能修剪叶子节点的父节点，修剪了父节点之后，叶子节点也会对应删掉

3.如果想在留下尽可能多的节点前提下，修剪掉所有的叶子节点。请你返回修剪后的二叉树。

有如下二叉树:

     o
    / \
   o   o
  / \  / \
 o  o o   o

修剪过后仅会留下根节点。

示例1

输入

{1,1,1,1,1,1,1}

输出

{1}

说明


叶子节点为最下面的4个1节点，但是不能直接修剪，只能修剪中间的2个1，修剪掉之后，只有根节点了

示例2

输入

{1,#,1,#,1,#,1,#,1}

输出

{1,#,1,#,1}

说明

退化为一条链了，将最后两个节点删除。

备注:

，删除根节点时返回为空。

说明：本题目包含复杂数据结构TreeNode，点此查看相关信息

算法知识视频讲解

Python

想去东北泡澡的共享单车人求你们别卷了

递归写不好写的3次遍历

1: 所有的叶节点值改为2

2: 所有叶节点的父节点值改为3

3: 所有值为3点节点改为None

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param root TreeNode类 
# @return TreeNode类
#
class Solution:
    def pruneLeaves(self , root ):
        # write code here

        def find(tree):
            if not tree.right and not tree.left:
                tree.val = 2 
            if tree.right:
                find(tree.right)
            if tree.left:
                find(tree.left)

        find(root)

        def find2(tree):
            if tree.right:
                if tree.right.val == 2:
                    tree.val = 3 
                else:
                    find2(tree.right)
            if tree.left:
                if tree.left.val == 2:
                    tree.val = 3 
                else:
                    find2(tree.left)

        find2(root) 

        def find3(tree):
            if tree.right:
                if tree.right.val != 1:
                    tree.right = None 
                else:
                    find3(tree.right) 
            if tree.left:
                if tree.left.val != 1:
                    tree.left = None 
                else:
                    find3(tree.left) 
        find3(root) 
        if root.val == 2&nbs***bsp;root.val == 3:
            return None 
        return root

发表于 2022-12-16 17:25:20 回复(5)