看题点击这里
这题思路是主要是后序遍历二叉树,需要注意的是在后序遍历(时间复杂度为)的时候将当前结点的父节点也要当作参数传入,如果没有就置为NULL,在遍历的过程中检查当前结点的val是不是在待删除的结点中。因为题中说明了树上每个节点都有一个不同的值都不一样,我先使用set集合将待删除的结点的val存储其中,时间复杂度为;再特判一下根节点是否在待删除的元素中若果要删除根节点,那么就不能将根节点加入到结果集中。最后开始后序遍历整棵树,在比较当前结点node的val是否在set集合中(set集合查找的时间复杂度为,如果存在则进行删除操作,首先是将node结点的非空子节点加入到结果集中,然后将当前结点的父节点指向该节点的指针置为NULL,需要注意的是判断一下pre结点是否为NULL,在不为NULL的情况下依次判断node结点是pre结点的左孩子还是右孩子,并将其左孩子指针或右孩子指针指向NULL。最后千万不要释放该节点的值(为什么我也不清楚,当时我释放了之后就提示地址错误)。
整个程序的时间复杂度为:。
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
class Solution {
private:
vector<TreeNode*> res;//结果集
set<int> s;//待删除结点集
public:
void postorder(TreeNode *node, TreeNode *pre) {
//当前结点为NULL
if (node == NULL) {
return;
}
postorder(node->left, node); //遍历左子树
postorder(node->right, node);//遍历右子树
if (s.count(node->val)) {//若果当前结点为待删除结点
if (node->left)//若果当前结点的左孩子结点不为空,则生成一颗新树
res.push_back(node->left);
if (node->right)//若果当前结点的右孩子结点不为空,则生成一颗新树
res.push_back(node->right);
if (pre != NULL) {//如果当前结点的父节点不为空
if (pre->left != NULL && pre->left->val == node->val) {//是父节点的左孩子
pre->left = NULL;//左孩子指向NULL
}
else if (pre->right != NULL && pre->right->val == node->val) {//是父节点的右孩子
pre->right = NULL;//右孩子指向NULL
}
}
/*
for(int i=0;i<3;i++)
printf("不能释放结点\n");
*/
//delete node;
}
}
vector<TreeNode*> delNodes(TreeNode* root, vector<int>& to_delete) {
for (int i = 0; i < to_delete.size(); i++) {//将待删除的结点添加进待删除节点集
s.insert(to_delete[i]);
}
//判断根节点是否是待删除结点
if (root != NULL && s.count(root->val) == 0) {
res.push_back(root);
}
postorder(root, NULL);//根节点的父节点为NULL,开始后序遍历
return res;
}
};
!点歌赞关注一下吧!