Advertisement
SalmaYasser

Untitled

Jan 15th, 2020
104
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.76 KB | None | 0 0
  1. set < int > del_set ;
  2. int sz = to_delete.size();
  3. for (int i = 0 ; i < sz ; i++)
  4. {
  5. del_set.insert(to_delete[i]);
  6. }
  7. vector < TreeNode* > res;
  8. fun (root , del_set, res, true);
  9. return res;
  10.  
  11. }
  12. TreeNode* fun (TreeNode* root, set<int>& to_delete, vector < TreeNode*>& res, bool r )
  13. {
  14. if (!root)
  15. return NULL;
  16.  
  17. bool del = to_delete.find(root->val) != to_delete.end();
  18.  
  19. if (r && !del)
  20. {
  21. res.push_back(root);
  22. }
  23.  
  24. root->left = fun (root -> left , to_delete, res, del);
  25. root->right = fun (root -> right , to_delete, res, del);
  26.  
  27. return del? NULL : root;
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement