题目描述
给定一颗二叉搜索树,请找出其中的第k大的结点。例如, 5 / \ 3 7 /\ /\ 2 4 6 8 中,按结点数值大小顺序第三个结点的值为4。
思路:二叉搜索树的一个重要性质就是它的中序遍历是排序的,因此这道题目只需要用中序遍历算法遍历一棵二叉搜索树,就很容易找出它的第K大结点。中序遍历用递归实现比较容易,但要想清楚的是遍历到一个根结点的时候要做的是什么?中序遍历二叉树打印的时候我们在递归完左子树之后打印根结点,本题目要求的当然不是打印,如果左子结点不是要找的结点,才会访问根结点,所以访问到根结点的时候要做的操作是将k减去1,因为左子结点已经证实不是要找的结点了,排除左子结点。这个过程可以看成目标移位的过程,每移过一个结点,K减1,直到K等于1时,当前结点就是要求的结点。
/* struct TreeNode { int val; struct TreeNode *left; struct TreeNode *right; TreeNode(int x) : val(x), left(NULL), right(NULL) { } }; */ class Solution { public: int c=0; TreeNode* KthNodecore(TreeNode* pRoot,int &k){//中序遍历 TreeNode* t=nullptr; if(pRoot->left){ t=KthNodecore(pRoot->left,k); } if(t) return t; c++; if(c==k) return pRoot; if(!t&&pRoot->right){ t=KthNodecore(pRoot->right,k); } if(t) return t; return nullptr; } /*TreeNode* target=nullptr; if(pRoot->left){//遍历左子树 target=KthNodecore(pRoot->left,k); } if(!target){ if(k==1) target=pRoot; k--; } if(!target&&pRoot->right)//遍历右子树 target=KthNodecore(pRoot->right,k); return target; }*/ TreeNode* KthNode(TreeNode* pRoot, int k) { if(!pRoot||k<1) return nullptr; return KthNodecore(pRoot,k); } };