पायथन में नोड्स हटाएं और फॉरेस्ट लौटाएं

मान लीजिए हमारे पास एक बाइनरी ट्री का रूट है, ट्री में प्रत्येक नोड का एक यूनिक मान है। to_delete में मान वाले सभी नोड्स को हटाने के बाद, हमें एक फॉरेस्ट बचा है। हमें शेष फॉरेस्ट में ट्रीज के रूट्स ढूंढने हैं। इसलिए यदि इनपुट इस तरह है

यदि to_delete ऐरे [3,5] जैसा है, तो आउटपुट होगा

इसे हल करने के लिए, हम इन चरणों का पालन करेंगे -

• res नामक एक ऐरे परिभाषित करें
• solve() नामक एक मेथड परिभाषित करें, यह node, to_delete ऐरे और एक बूलियन टाइप info लेगा जो बता रहा है कि नोड रूट है या नहीं। मेथड नीचे की तरह कार्य करेगा -
• यदि node null है, तो null लौटाएं
• flag := true यदि नोड का मान to_delete ऐरे में है
• यदि flag false है और is_root true है, तो node को res में इन्सर्ट करें
• node का बायां := solve(node का बायां, to_delete, flag)
• node का दायां := solve(node का दायां, to_delete, flag)
• flag सेट होने पर none लौटाएं, अन्यथा false
• मुख्य मेथड से solve() मेथड को इस तरह कॉल करें solve(node, to_delete, true)

उदाहरण

class Solution(object):
    def delNodes(self, root, to_delete):
        \"\"\"
        :type root: TreeNode
        :type to_delete: List[int]
        :rtype: List[TreeNode]
        \"\"\"
        to_delete = set(to_delete)
        self.res = []
        self.solve(root,to_delete,True)
        return self.res
    def solve(self,node,to_delete,is_root):
        if not node:
            return None
        flag = node.val in to_delete
        if not flag and is_root:
            self.res.append(node)
        node.left = self.solve(node.left,to_delete,flag)
        node.right = self.solve(node.right,to_delete,flag)
        return None if flag else node

इनपुट

[1,2,3,4,5,6,7] [3,5]

आउटपुट

[[1,2,null,4],[6],[7]]