1110. 删点成林

题目具体描述参见leetcode网站

主要分享内容，层序遍历题解方法

题目描述：

给出二叉树的根节点 root，树上每个节点都有一个不同的值。 如果节点值在 to_delete 中出现，我们就把该节点从树上删去，最后得到一个森林（一些不相交的树构成的集合）。 返回森林中的每棵树。你可以按任意顺序组织答案。

题目解读

除根节点外，只有当找到要删除的节点时才可能出现新的树 示例图如下

c++ 代码实现

/**
 * 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*> delNodes(TreeNode* root, vector<int>& to_delete) {
        if(to_delete.size() == 0) return vector<TreeNode*>{root};
        unordered_set<int> st(to_delete.begin(), to_delete.end());
        // 构造一定需要删除的虚拟head节点，保证当根节点需要作为答案一员时，能统一处理
        TreeNode* head = new TreeNode();
        head->left = root;
        head->val = to_delete[0];
        queue<TreeNode*> que;
        que.push(head);
        vector<TreeNode*> ans;
        while(que.size() != 0){
            auto node = que.front(); que.pop();
            if(node == nullptr) continue;
            // 层序遍历树，因此将左右节点放入队列中
            que.push(node->left);
            que.push(node->right);
            // 如果需要删除，才尝试放入其左右节点
            if(st.count(node->val) != 0){
                // 如果存在子节点，且子节点是不被删除的，则存入，否则不存入
                if(node->left && st.count(node->left->val) == 0){
                    ans.push_back(node->left);
                }
                if(node->right && st.count(node->right->val) == 0){
                    ans.push_back(node->right);
                }
            }else{
                // 如果不需要删除，则不做任何放入的操作, 但当其子节点要被删除，可以断开当前和其子节点的关系
                if(node->left && st.count(node->left->val) != 0){
                    node->left = nullptr;
                }
                if(node->right && st.count(node->right->val) != 0){
                    node->right = nullptr;
                }
            }
        }
        return ans;
    }
};