牛客題霸 [二叉搜索樹的第k個結(jié)點(diǎn)]C++題解/答案

題目：

給定一棵二叉搜索樹，請找出其中的第k小的結(jié)點(diǎn)。

題解：

二叉搜索樹： 若它的左子樹不空，則左子樹上所有結(jié)點(diǎn)的值均小于它的根結(jié)點(diǎn)的值; 若它的右子樹不空，則右子樹上所有結(jié)點(diǎn)的值均大于它的根結(jié)點(diǎn)的值

我們寫出二叉搜索樹的中序遍歷，就是2345678.其實(shí)是順序排列的，那我們要找第k小的值，在中序遍歷的過程中找第k個元素
nodeCount是探究root的左子樹里有幾個節(jié)點(diǎn)，因?yàn)槭琼樞蚺帕?#xff0c;如果root>k，就說明k在左子樹里面，遞歸左子樹就行。如果root<k，就說明k在右子樹里面，遞歸右子樹就行，遞歸右子樹時，第k小就改成第 k - node_count-1小（因?yàn)橛易訕淙茸笞訕浜透?#xff09;

代碼：

/* struct TreeNode {int val;struct TreeNode *left;struct TreeNode *right;TreeNode(int x) :val(x), left(NULL), right(NULL) {} }; */ class Solution { public:TreeNode* KthNode(TreeNode* pRoot, int k){if(!pRoot) return nullptr;int node_count = nodeCount(pRoot->left);if (k <= node_count)return KthNode(pRoot->left, k);if (k == node_count + 1)return pRoot;return KthNode(pRoot->right, k - node_count-1);}int nodeCount(TreeNode* root) {if (!root)return 0;return nodeCount(root->left) + nodeCount(root->right) + 1;} };

總結(jié)

以上是生活随笔為你收集整理的牛客题霸 [二叉搜索树的第k个结点]C++题解/答案的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。