【BST】在搜索二叉树中,寻找节点数据域在[L,U]范围内的节点

题目:EPI


书上代码的思路是:由于BST的中序遍历的节点是按照升序排列的,所以先找到第一个(按照中序遍历的顺序)大于等于L的节点,然后依次找出这个节点的后继节点中,落在范围[L,U]的节点。

但有一个问题就是,如果二叉树节点没有指向父节点的指针域,这个方法就不可取。何况题目没有要求返回的节点要按照升序排列。所以我写出以下代码,不需要二叉树节点有指向父节点的指针,返回的结果在[L,U]范围,但不是有序的。

void find_larger_or_equal_than_L(const shared_ptr<treenode> root, const int L, vector<shared_ptr<treenode>> &res)
{
	if (root == nullptr)
		return;
	if (root->data < L)
	{
		find_larger_or_equal_than_L(root->right, L, res);
		return;
	}
	res.push_back(root);
	find_larger_or_equal_than_L(root->left, L, res);
	find_larger_or_equal_than_L(root->right, L, res);	
}

void find_smaller_or_equal_than_U(const shared_ptr<treenode> root, const int U, vector<shared_ptr<treenode>> &res)
{
	if (root == nullptr)
		return;
	if (root->data >U)
	{
		find_smaller_or_equal_than_U(root->left, U, res);
		return;
	}
	res.push_back(root);
	find_smaller_or_equal_than_U(root->left, U, res);
	find_smaller_or_equal_than_U(root->right, U, res);
}

vector<shared_ptr<treenode>> find_range_nodes(const shared_ptr<treenode> root, const int L, const int U)
{
	vector<shared_ptr<treenode>> res;
	if (root == nullptr || L>U)
		return res;
	//找一个范围在[L,U]的节点
	shared_ptr<treenode> cur = root;
	while (cur)
	{
		if (cur->data < L)
			cur = cur->right;
		else if (cur->data > U)
			cur = cur->left;
		else
			break;
	}
	if (cur == nullptr)
		return res;
	res.push_back(cur);//cur符合要求, 插入res
	find_larger_or_equal_than_L(cur->left, L,res);//cur的左子树中大于等于L的节点插入res
	find_smaller_or_equal_than_U(cur->right, U, res);//cur的右子树中小于等于U的节点插入res
	return res;
}


  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值