二叉树的下一个节点:无需线索化,直接找规律。

题目描述

给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。


思路:抓住中序遍历的本质,左根右,找规律即可。

解法一:

分析二叉树的下一个节点,一共有以下情况:
1.二叉树为空,则返回空;
2.节点右孩子存在,则设置一个指针从该节点的右孩子出发,一直沿着指向左子结点的指针找到的叶子节点即为下一个节点;
3.节点不是根节点。如果该节点是其父节点的左孩子,则返回父节点;否则继续向上遍历其父节点的父节点,重复之前的判断,返回结果。代码如下:
我把他更加细化了,其实可以只分3个讨论:

/*
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==NULL) return NULL;//1.空节点
        else if(pNode->right){//2.只要右孩子节点存在
            TreeLinkNode*p=pNode->right;
            while(p->left) p=p->left;
            return p;
        }else if(pNode->left)//3.右孩子不存在,左孩子存在
            return pNode->next;
        else{//4.左右孩子均不存在,即叶节点。第3点可以和第4点融合,因为都是往左上回溯。
            if(pNode->next==NULL){//这条容易忽略,因为该叶子节点可能为根节点,造成访问NULL->next的错误。
                return NULL;
            }
            else if(pNode->next->left==pNode){//是某父亲的左孩子,则直接返回父亲。
                return pNode->next;
            }else{
                TreeLinkNode*p=pNode;//是某父亲的右孩子,则“往左上”回溯,找规律即可。
                while(p->next&&p->next->right==p) p=p->next;//
                return p->next;
            }
        }
    }
};


解法二:二叉树的线索话,设置一个全局尾节点。

/*
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)
    {
        TreeLinkNode*root=pNode;
        if(root==NULL) return NULL;
        while(root->next) root=root->next;

        TreeLinkNode*pLastNodeInList=NULL;
        ConvertNode(root,&pLastNodeInList);
        return pNode->right;
    }

    void ConvertNode(TreeLinkNode*pNode,TreeLinkNode**pLastNodeInList){//二叉树的线索化,用一个指针变量指着一个固定地址,这个固定地址里的值是链表的尾节点。
        if(pNode==NULL) 
            return ;
        TreeLinkNode*cur=pNode;
        if(cur->left)
            ConvertNode(cur->left,pLastNodeInList);//中序遍历,原地调整,二叉树的线索化。
        cur->left=*pLastNodeInList;
        if(*pLastNodeInList!=NULL)
            (*pLastNodeInList)->right=cur;//*()->要打括号。不能直接写*pLastNodeInList->right=cur;,这是错误的,优先级。
        *pLastNodeInList=cur;
        if(cur->right)
            ConvertNode(cur->right,pLastNodeInList);
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值