1. 程式人生 > >62、劍指offer--二叉搜索樹的第k個結點

62、劍指offer--二叉搜索樹的第k個結點

pro clas -- || nbsp oot 其中 ret 解題思路

題目描述 給定一顆二叉搜索樹,請找出其中的第k大的結點。例如, 5 / \ 3 7 /\ /\ 2 4 6 8 中,按結點數值大小順序第三個結點的值為4。 解題思路:使用中序遍歷進行遍歷,得到的就是按照順序的,當遍歷到第k個就輸出即可。
 1 /*
 2 struct TreeNode {
 3     int val;
 4     struct TreeNode *left;
 5     struct TreeNode *right;
 6     TreeNode(int x) :
 7             val(x), left(NULL), right(NULL) {
 8     }
9 }; 10 */ 11 class Solution { 12 public: 13 //實現中序遍歷,對於二叉搜索樹中序遍歷是按節點從小到大排序 14 TreeNode *KthNodeCore(TreeNode *root, int &k)//註意此處k為引用,因為在操作過程中k改變 15 { 16 TreeNode *target = NULL; 17 if(root->left != NULL) 18 { 19 target = KthNodeCore(root->left,k);
20 } 21 if(target == NULL)//無左節點且還未找到,返回其葉子節點的父節點,k-- 22 { 23 if(k == 1) 24 target = root; 25 k--; 26 } 27 if(target == NULL && root->right != NULL)//找右結點 28 { 29 target = KthNodeCore(root->right,k);
30 } 31 return target; 32 } 33 TreeNode* KthNode(TreeNode* pRoot, int k) 34 { 35 if(pRoot == NULL || k == 0) 36 return NULL; 37 return KthNodeCore(pRoot,k); 38 } 39 };

62、劍指offer--二叉搜索樹的第k個結點