leetcode 144周赛 删点成林

3 篇文章 0 订阅
2 篇文章 0 订阅

看题点击这里

        这题思路是主要是后序遍历二叉树,需要注意的是在后序遍历(时间复杂度为O(n))的时候将当前结点的父节点也要当作参数传入,如果没有就置为NULL,在遍历的过程中检查当前结点的val是不是在待删除的结点中。因为题中说明了树上每个节点都有一个不同的值都不一样,我先使用set集合将待删除的结点的val存储其中,时间复杂度为O(n);再特判一下根节点是否在待删除的元素中若果要删除根节点,那么就不能将根节点加入到结果集中。最后开始后序遍历整棵树,在比较当前结点node的val是否在set集合中(set集合查找的时间复杂度为O(log n),如果存在则进行删除操作,首先是将node结点的非空子节点加入到结果集中,然后将当前结点的父节点指向该节点的指针置为NULL,需要注意的是判断一下pre结点是否为NULL,在不为NULL的情况下依次判断node结点是pre结点的左孩子还是右孩子,并将其左孩子指针或右孩子指针指向NULL。最后千万不要释放该节点的值(为什么我也不清楚,当时我释放了之后就提示地址错误)。

整个程序的时间复杂度为:n+n*logn=nlogn

  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;
	}
};

!点歌赞关注一下吧!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值