剑指 Offer(第2版)面试题 54:二叉搜索树的第 k 大节点

剑指 Offer(第2版)面试题 54:二叉搜索树的第 k 大节点

  • [剑指 Offer(第2版)面试题 54:二叉搜索树的第 k 大节点](#剑指 Offer(第2版)面试题 54:二叉搜索树的第 k 大节点)

剑指 Offer(第2版)面试题 54:二叉搜索树的第 k 大节点

题目来源:

  1. AcWing 70. 二叉搜索树的第k个结点
  2. LeetCode 230. 二叉搜索树中第K小的元素

解法1:中序遍历 - 辅助数组

二叉搜索树(BST)的中序遍历序列是递增的,中序遍历得到节点序列后,返回 nodes[k - 1]。

代码:

c 复制代码
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution
{
public:
	TreeNode *kthNode(TreeNode *root, int k)
	{
		if (root == nullptr)
			return nullptr;
		vector<TreeNode *> nodes;
		inOrder(root, nodes);
		return nodes[k - 1];
	}
	void inOrder(TreeNode *root, vector<TreeNode *> &nodes)
	{
		if (root == nullptr)
			return;
		inOrder(root->left, nodes);
		nodes.push_back(root);
		inOrder(root->right, nodes);
	}
};

复杂度分析:

时间复杂度:O(logn),其中 n 是二叉搜索树的节点个数。

空间复杂度:O(n),其中 n 是二叉搜索树的节点个数。

解法2:中序遍历

好像没有必要得到整个中序遍历序列,我们设置一个全局变量,值与 k 相同。

在中序遍历中,每经过一个节点,全局变量减 1,当全局变量为 0 时,当前节点就是答案。

剪枝:当 k <= 0 时,就没必要再递归下去了。

代码:

c 复制代码
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution
{
private:
	TreeNode *ans;

public:
	TreeNode *kthNode(TreeNode *root, int k)
	{
		if (root == nullptr || k == 0)
			return nullptr;
		inOrder(root, k);
		return ans;
	}
	void inOrder(TreeNode *root, int &k)
	{
		if (root == nullptr)
			return;
		inOrder(root->left, k);
		k--;
		if (k == 0)
		{
			ans = root;
			return;
		}
		if (k > 0)
			inOrder(root->right, k);
	}
};

复杂度分析:

时间复杂度:O(height + k),其中 height 是二叉搜索树的高度。

空间复杂度:O(1)。

相关推荐
LyaJpunov9 天前
深入理解 C++ volatile 与 atomic:五大用法解析 + 六大高频考点
c++·面试·volatile·atomic
小灰灰搞电子9 天前
Qt PyQt与PySide技术-C++库的Python绑定
c++·qt·pyqt
时空自由民.9 天前
C++ 不同线程之间传值
开发语言·c++·算法
Ray_19979 天前
C++二级指针的用法指向指针的指针(多级间接寻址)
开发语言·jvm·c++
双叶8369 天前
(C语言)Map数组的实现(数据结构)(链表)(指针)
c语言·数据结构·c++·算法·链表·哈希算法
Jay_5159 天前
C++ STL 模板详解:由浅入深掌握标准模板库
c++·学习·stl
Cyrus_柯9 天前
C++(面向对象编程——继承)
开发语言·c++·算法·面向对象
Echo``9 天前
12.OpenCV—基础入门
开发语言·c++·人工智能·qt·opencv·计算机视觉
十秒耿直拆包选手9 天前
C++:动态库相关文件
c++
小L~~~10 天前
C++网络编程入门学习(五)-- CMake 学习笔记
linux·c++