二叉树的右视图,二叉搜索树中第k小的元素

第一题:

代码如下:

vector<int> rightSideView(TreeNode* root) 
{
	vector<int> res;
	if (root == nullptr)
		return res;
	queue<TreeNode*> q;
	q.push(root);
	int k = 1;
	while (!q.empty())
	{
		while (k--)
		{
			TreeNode* front = q.front();
			q.pop();
			if (k == 0)
				res.push_back(front->val);
			if (front->left)
				q.push(front->left);
			if (front->right)
				q.push(front->right);
		}
		k = q.size();
	}
	return res;
}

思路整理:其实可以转化为找到二叉树的每层的最后一个元素,本质就是一个层序遍历,借助数据结构队列来解决问题即可。

具体逻辑及变量解释:

<1>res表示最终结果,q为队列,k表示该层的节点数。

<2>将根节点放入队列,当队列不为空时,取出队列的第一个节点,放入该根节点的左孩子节点与右孩子节点,当k == 0 时,表示此时的节点是该层最后一个节点,将结果放入res即可,当该层出完后,队列中的元素个数就是下一层的节点数。

<3>返回res即可。

第二题:

代码如下:

class Solution 
{
public:
	int count;
	int res = 0;
	int kthSmallest(TreeNode* root, int k) 
	{
		count = k;
		_dfs(root);
		return res;
	}
	void _dfs(TreeNode* root)
	{
		if (root == nullptr)
			return;
		_dfs(root->left);
		if (--count == 0)
		{
			res = root->val;
			return;
		}
		_dfs(root->right);
	}
};

思路整理:二叉搜索树的中序遍历是有序的,因此只需要一个全局变量来记录目前到达了有序序列的第几个元素即可得到最终答案。

具体逻辑以及变量解释:

<1>res代表最终结果,count代表计数。

<2>使用中序遍历即可,如果count减到0,就代表找到了结果,返回即可。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值