二叉树 寻找二叉搜索树第K大的节点

给定一颗二叉搜索树,返回该二叉搜索树第K大的节点

思路:由于二叉搜索树的中序遍历序列是从小到大的序列,模拟这一过程每访问一个节点,count++直到第K个

注意点:

  • 由于递归的进入与返回过程,递归调用栈频繁进出,计数器可设置成全局的变量
  • 递归的返回条件是NULL 注意特殊情况
class Solution {
public:
    int count=0;//计数器  注意设置成全局,避免放入递归调用栈
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if(pRoot==NULL)
            return NULL;
        {
            TreeNode* left=KthNode(pRoot->left,k);
            if(left!=NULL)
                return left;
            count++;
            if(count==k)
                return pRoot;
            TreeNode* right=KthNode(pRoot->right,k);
            if(right!=NULL)
                return right;  
        }
        return NULL;//没有找到,可能是K过大
    } 
    
};

变种:求倒数第K大的节点

class Solution {
public:
    int count=0;//计数器  注意设置成全局,避免放入递归调用栈
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if(pRoot==NULL)
            return NULL;
        {
            TreeNode* right=KthNode(pRoot->right,k);
            if(left!=NULL)
                return right;
            count++;
            if(count==k)
                return pRoot;
            TreeNode* left=KthNode(pRoot->left,k);
            if(left!=NULL)
                return left;  
        }
        return NULL;//没有找到,可能是K过大
    }
   

    
};

思路:类似于中序遍历,只要将左右子树的访问顺序倒置即可

猜你喜欢

转载自blog.csdn.net/qq_33369979/article/details/88170462