1110. 刪點成林
阿新 • • 發佈:2021-06-20
給出二叉樹的根節點root,樹上每個節點都有一個不同的值。
如果節點值在to_delete中出現,我們就把該節點從樹上刪去,最後得到一個森林(一些不相交的樹構成的集合)。
返回森林中的每棵樹。你可以按任意順序組織答案。
示例:
輸入:root = [1,2,3,4,5,6,7], to_delete = [3,5]
輸出:[[1,2,null,4],[6],[7]]
提示:
樹中的節點數最大為1000。
每個節點都有一個介於1 到1000之間的值,且各不相同。
to_delete.length <= 1000
to_delete 包含一些從1 到1000、各不相同的值。
/** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode() : val(0), left(nullptr), right(nullptr) {} * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} * TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {} * }; */ class Solution { public: vector<TreeNode*> res; set<int> delSet; TreeNode* delNodesProcess(TreeNode* root) { if(root == nullptr) { return nullptr; } root->left = delNodesProcess(root->left); root->right = delNodesProcess(root->right); if(delSet.find(root->val) != delSet.end()) { if(root->left !=nullptr) { res.push_back(root->left); } if(root->right !=nullptr) { res.push_back(root->right); } root = nullptr; } return root; } vector<TreeNode*> delNodes(TreeNode* root, vector<int>& to_delete) { if(root == nullptr) { return res; } int len = to_delete.size(); for(int i = 0; i < len; i++) { delSet.insert(to_delete[i]); } if(root == nullptr) { return res; } root->left = delNodesProcess(root->left); root->right = delNodesProcess(root->right); if(delSet.find(root->val) != delSet.end()) { if(root->left !=nullptr) { res.push_back(root->left); } if(root->right !=nullptr) { res.push_back(root->right); } } else { res.push_back(root); } return res; } };