C++刷LeetCode（1110. 删点成林）【树】【递归】

糖逗

题目描述：

1. 给出二叉树的根节点 root，树上每个节点都有一个不同的值。
   
2. 
   
3. 如果节点值在 to_delete 中出现，我们就把该节点从树上删去，最后得到一个森林（一些不相交的树构成的集合）。
   
4. 
   
5. 返回森林中的每棵树。你可以按任意顺序组织答案。
   
6. 
   
7.  
   
8. 
   
9. 示例：
   
10. 
    
11. 
    
12. 
    
13. 输入：root = [1,2,3,4,5,6,7], to_delete = [3,5]
    
14. 输出：[[1,2,null,4],[6],[7]]
    
15.  
    
16. 
    
17. 提示：
    
18. 
    
19. 树中的节点数最大为 1000。
    
20. 每个节点都有一个介于 1 到 1000 之间的值，且各不相同。
    
21. to_delete.length <= 1000
    
22. to_delete 包含一些从&#160;1 到&#160;1000、各不相同的值。
    
23. 
    
24. 来源：力扣（LeetCode）
    
25. 链接：https://leetcode-cn.com/problems/delete-nodes-and-return-forest
    
26. 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

复制代码

1. class Solution {
   
2. public:
   
3.     void recursion(TreeNode*& root, vector<int> to_delete, vector<TreeNode*>& res){
   
4.         if(root == NULL)return;
   
5.         recursion(root -> left, to_delete, res);
   
6.         recursion(root -> right, to_delete, res);
   
7.         if(std::find(to_delete.begin(), to_delete.end(), root -> val) != to_delete.end()){
   
8.             if(root -> left)res.push_back(root -> left);
   
9.             if(root -> right)res.push_back(root -> right);
   
10.             root = NULL;
    
11.         }
    
12.     }
    
13.     vector<TreeNode*> delNodes(TreeNode* root, vector<int>& to_delete) {
    
14.         vector<TreeNode*> res;
    
15.         recursion(root, to_delete, res);
    
16.         if(root != NULL)res.push_back(root);
    
17.         return res;
    
18.     }
    
19. };

复制代码