二叉搜索树的下一个结点

最新推荐文章于 2024-10-01 16:50:15 发布

原创最新推荐文章于 2024-10-01 16:50:15 发布 · 363 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

算法专栏收录该内容

77 篇文章

订阅专栏

首先这题是有两种变形的，第一种TreeNode包含一个指向父节点的指针，然后直接给你一个节点，然你给出其下一个终须遍历节点的位置：

/*
struct TreeLinkNode {
    int val;
    struct TreeLinkNode *left;
    struct TreeLinkNode *right;
    struct TreeLinkNode *next;
    TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {

    }
};
*/
class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        if(!pNode)
            return NULL;
        if(pNode->right){
            pNode = pNode->right;
            while(pNode->left)
                pNode = pNode->left;
            return pNode;
        }else{
            while(pNode->next && pNode->next->left != pNode) //注意需要先判断是否是空指针
                pNode = pNode->next;
            return pNode->next;
        }
    }
};

还有一种是直接给你一个二叉树根节点t以及一个节点p，让你在二叉树中找出这个节点的下一个节点，如下所示：

//这个稍微有点抽象
/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/

class Successor {
private:
    TreeNode * pre = new TreeNode(-1); //为了不让他等于p
public:
    int findSucc(TreeNode* root, int p) {
        if(root == NULL)
            return -1;
        int curr = findSucc(root->left, p);
        if(curr == -1){//已经是当前的最左节点
            if(p == pre->val)
                return root->val;
            pre = root;
            return findSucc(root->right, p);
        }
        return curr;
    }
};

或者是下面这种形式的

class Successor {
private:
    TreeNode * pre = new TreeNode(-1); //为了不让他等于p
public:
    int findSucc(TreeNode* root, TreeNode* p) {
        if(root == NULL)
            return NULL;
        TreeNode * curr = findSucc(root->left, p);
        if(curr == NULL){//root现在已经是当前的最左节点
            if(p == pre)
                return root;
            pre = root;
            return findSucc(root->right, p);
        }
        return curr;
    }
};