/*
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||k<=0)
return nullptr;
return judgeknode(pRoot,k);
}
private:
int knode=0;
TreeNode* judgeknode(TreeNode* pRoot,int k){
TreeNode* ansptr;
if(pRoot->left!=nullptr)
ansptr = judgeknode(pRoot->left,k);
if(ansptr!=nullptr)
return ansptr;
++knode;
if(knode==k)
return pRoot;
if(pRoot->right!=nullptr)
ansptr = judgeknode(pRoot->right,k);
return ansptr;
}
};
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||k<=0)
return nullptr;
return judgeknode(pRoot,k);
}
private:
int knode=0;
TreeNode* judgeknode(TreeNode* pRoot,int k){
TreeNode* ansptr;
if(pRoot->left!=nullptr)
ansptr = judgeknode(pRoot->left,k);
if(ansptr!=nullptr)
return ansptr;
++knode;
if(knode==k)
return pRoot;
if(pRoot->right!=nullptr)
ansptr = judgeknode(pRoot->right,k);
return ansptr;
}
};