C++ 寻找下一个结点

#include<iostream>
using namespace std;
#include<vector>

/**
请设计一个算法，寻找二叉树中指定结点的下一个结点
（即中序遍历的后继）。

给定树的根结点指针TreeNode* root和结点的值int p，
请返回值为p的结点的后继结点的值。
保证结点的值大于等于零小于等于100000且没有重复值，若不存在后继返回-1。

二叉树中的每个节点都有父亲节点(除了根节点),给定当前节点curr,如何最快找到
当前节点的父亲节点,对二叉树进行深度优先遍历或者广度优先遍历,遍历过程中
指针指向的是当前遍历得到的节点temp,判断如果当前节点temp的左孩子或者当前节点temp的右孩子
等于curr节点,则返回temp节点就是curr节点的父亲节点

二叉树的深度优先遍历包括:先序遍历、中序遍历和后序遍历
二叉树的深度优先遍历是通过栈结构实现的
二叉树后序遍历的非递归/迭代实现过程:
由于是先后序遍历左子树,再后序遍历右子树,最后访问根节点
故而压入栈中的顺序是: 根节点,右孩子节点,左孩子节点

递归算法实际上是调用了系统栈，二叉树的深度优先遍历的非递归形式使用栈
实现,实际上就是用函数定义的栈代替了递归算法过程中调用的系统栈

二叉树的广度优先遍历包括:二叉树的层序遍历
二叉树的广度优先遍历是通过队列结构实现的

**/

struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};

class Successor {
private:
    vector<int> seq;
public:
    void mid_dfs(TreeNode* root){
        if(root==NULL){
            return;
        }
        if(root->left==NULL and root->right==NULL){
            seq.push_back(root->val);
            return;
        }
        if(root->left!=NULL){
            this->mid_dfs(root->left);
        }
        seq.push_back(root->val);
        if(root->right!=NULL){
            this->mid_dfs(root->right);
        }
        return;
    }
    int findSucc(TreeNode* root, int p) {
        /**
        算法:直接对二叉树进行中序遍历,得到中序遍历序列后的下一个数值
        中序遍历所需要的输出序列是:
        先对左子树进行中序遍历,再访问根节点,最后中序遍历右子树
        借助栈的数据结构,
        **/
        this->mid_dfs(root);
        unsigned int index=0;
        for(unsigned int i=0;i<seq.size();i++){
            if(seq[i]==p){
                index=i;
                break;
            }
        }
        if(index==seq.size()-1){
            return -1;
        }
        else{
            return seq[index+1];
        }
    }
};

int main(){
    TreeNode *root=new TreeNode(1);
    root->left=new TreeNode(2);
    root->right=new TreeNode(3);

    root->left->left=new TreeNode(4);
    root->left->right=new TreeNode(5);

    root->right->left=new TreeNode(6);
    root->right->right=new TreeNode(7);

    root->left->left->left=new TreeNode(8);
    root->left->left->right=new TreeNode(9);

    root->left->right->left=new TreeNode(10);
    root->left->right->right=new TreeNode(11);

    root->right->left->left=new TreeNode(12);
    root->right->left->right=new TreeNode(13);

    root->right->right->left=new TreeNode(14);
    root->right->right->right=new TreeNode(15);

    Successor a;
    int result;
    result=a.findSucc(root,11);
    cout<<result<<endl;
    return 0;
}