剑指offer-08 二叉树的下一个节点

剑指offer第8题,本来想找leetcode上对应的题,后来没找到,直接去牛客网上刷了。

题目描述:

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

分析

我看到这道题的第一个想法,就是不用管左子树。因为中序遍历,左子树是之前访问过了,所以下一个节点只有两种可能。
一、在右子树
二、在父亲节点的部分
分情况讨论:
一、在右子树
如果右子树只有一个节点那还好说,那么下一个节点就是右子节点。但是如果右子树是一个树,那么就得看右子树的最左节点。

        TreeLinkNode * point=pNode->right;
        while(point->left!=NULL)
        {
            point=point->left;
        }
        return point;

二、在父亲节点之上
若右子树为空,此时分两种情况。
(1)pNode是父亲节点的左子节点,那正好,下一个要访问的就是父亲节点,直接返回父亲节点就行;
(2)pNode是父亲节点的右子节点,说明下一个节点所在的区域,还至少在父亲节点的父亲节点之上。
我一开始绕了个远,第二种情况,我想到的是用递归解决。如果pNode是父亲节点的右子节点,那么就pNode=pNode->next;同时砍去父亲节点的右子树。这样就构造了一个子问题,把问题转换成了给定父亲节点之后,寻找访问下一个节点的问题。

        if(pNode->right==NULL)
        {
            if(pNode->next==NULL||pNode==pNode->next->left)
                return pNode->next;
            else{
                pNode->next->right=NULL;
                return GetNext(pNode->next);
            }
        }

说的有点绕,整体思想就是砍去右子树,让指针指向父亲,然后在新的树里,寻找指针指向的节点要访问的下一个节点。其实我个人觉得还挺巧妙的。
不过还是把简单问题复杂化了,其实直接一直往父亲寻找就行,直到找到满足情况(1)的pNode,返回pNode->next。

        if(pNode->right==NULL)
        {
            while(pNode->next!=NULL&&pNode!=pNode->next->left)
                pNode=pNode->next;
            return pNode->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)
    {
        if(pNode==NULL)
            return NULL;
        if(pNode->right==NULL)
        {
            while(pNode->next!=NULL&&pNode!=pNode->next->left)
                pNode=pNode->next;
            return pNode->next;
        }
        TreeLinkNode * point=pNode->right;
        while(point->left!=NULL)
        {
            point=point->left;
        }
        return point;
        
    }
};
posted @ 2019-09-18 21:19  HaoPeng_Zhang  阅读(47)  评论(0编辑  收藏