61二叉搜索树的第k个结点

题目描述

给定一颗二叉搜索树，请找出其中的第k大的结点。例如， 5 / \ 3 7 /\ /\ 2 4 6 8 中，按结点数值大小顺序第三个结点的值为4。

思路：二叉搜索树的中序遍历是递增的序列，使用循环的中序遍历找到第k个节点就行了，对中序遍历的循环版本没理解，使用一个stack，找到最左边的节点，所以初始化的时候首先初始化节点，stack要为空。计数的cnt要初始化为0。不清楚的话就一个节点的时候判断一下，然后p = p->right，不需要判断为空，因为为空的话，上面的循环不会进入。

/*
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 == nullptr){return nullptr;}stack<TreeNode*> s;  //s.push(pRoot);TreeNode* p = pRoot;int cnt = 0;while(!s.empty() || p != nullptr){while(p != nullptr){s.push(p);p = p->left;}if(!s.empty()){p = s.top();s.pop();                ++cnt;if(cnt == k){return p;}                 p = p->right; }}return nullptr;}    
};