Re: [閒聊] 每日leetcode

[sustainer123 (caster )]

※ 引述《Rushia (早瀬ユウカの体操服 )》之銘言：
: https://leetcode.com/problems/delete-leaves-with-a-given-value/description/
: 1325. Delete Leaves With a Given Value
: 給你一個二元樹和一個數字 target，刪除所有葉子節點是 target 的節點，如果刪除後
: 新的葉子節點也是 target 也要刪除。
: 思路：
: 1.如果遇到葉子節點且是target就刪除他並返回null，每次檢查之間要先dfs，如果dfs
:   返回null就刪除子節點。
: py code:
: ---------------------------------------------
: class Solution:
:     def removeLeafNodes(self, root: Optional[TreeNode], target: int) ->
: Optional[TreeNode]:
:         if not root:
:             return None
:         root.left = self.removeLeafNodes(root.left, target)
:         root.right = self.removeLeafNodes(root.right, target)
:         if not root.left and not root.right and root.val == target:
:             return None
:         return root
: ---------------------------------------------

思路:
差不多 其實就DFS然後找到葉節點且符合目標值就刪除

Python Code:
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def removeLeafNodes(self, root: Optional[TreeNode], target: int) ->
Optional[TreeNode]:
        def dfs(node):
            if not node :
                return None
            node.right = dfs(node.right)
            node.left = dfs(node.left)

            if not node.left and not node.right and node.val == target:
                return None
            return node

        return dfs(root)

--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 114.43.131.26 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/Marginalman/M.1715926470.A.F98.html
→ JIWP: 別卷了
→ digua: 大師
→ wu10200512: 我要被你卷死了
→ sustainer123: 你都上岸了 對啊
推 Smallsh: 泥板剩我DFS還要翻課本抄了
推 cities516: 別卷了
推 SecondRun: 大師 救我
→ DJYOSHITAKA: 佬 教
→ sustainer123: 我文組廢物
推 JIWP: 你是下一個姑姑魯
推 steven183: 別卷了