第一题:
代码如下:
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,就代表找到了结果,返回即可。