JZ8 二叉树的下一个结点

1. 问题描述

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


注意：该题的输入只有一个结点。该结点不是指向根节点，而是指向给定的那个结点。

2. 解法一：简单

算法思想

1. 根据给出的结点求出整棵树的根节点
2. 根据根节点递归求出树的中序遍历，存入vector
3. 在vector中查找当前结点，则当前结点的下一结点即为所求。

代码

/*
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:
    void pre_order(TreeLinkNode *root, vector<TreeLinkNode*> &v) {
        if (!root)  return;
        pre_order(root->left, v);
        v.push_back(root);
        pre_order(root->right, v);
    }
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
          TreeLinkNode *root = nullptr;
          TreeLinkNode *tmp = pNode;
          while (tmp) {  // 第一步
              root = tmp;
              tmp = tmp->next;
          }   
          vector<TreeLinkNode*> v;
          pre_order(root, v);  // 第二步
          int n = v.size();
          for (int i = 0; i < n; ++i) { // 第三步
              if (v[i] == pNode && i + 1 != n) return v[i+1];
          }
          return nullptr;
    }
};

时间空间复杂度分析

时间复杂度：O(N)
空间复杂度：O(N)

3. 解法二：巧妙

算法思想

如图，下一个节点分三种情况
（1）对于2、3、6，其右儿子存在。则一直找该右儿子的左孩子为空的那个结点。
（2）对于1、4、5，其右儿子不存在。则找其父亲结点。如果父亲结点的左孩子就是该结点，则父亲结点就是该结点的下一个结点。否则循环找父亲结点的父亲结点，接着刚刚的判断。
（3）对于7，第二步以pNode->next==nullptr结束，则说明没有比其更大的结点。返回nullptr。

代码

/*
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 pNode;
        if (pNode->right) {  // 属于[2 3 6]类
            pNode = pNode->right;
            while (pNode->left) pNode = pNode->left;
            return pNode;
        }
        while (pNode->next) {   // 属于 [1] 和 [4 5]
            TreeLinkNode *root = pNode->next;
            if (root->left == pNode)  return root;
             pNode = pNode->next;
        }
        return nullptr;   // 属于[7]
    }
};

时间空间复杂度分析

时间复杂度：最坏情况下为O(N)
空间复杂度：O(1)