Inorder Successor in BST

Given a binary search tree and a node in it, find the in-order successor of that node in the BST.

在树中寻找一个节点在中序遍历下的后续节点，此类问题是带条件的遍历，设定某个条件，在遍历过程中符合给定条件时，将节点存入结果，最终返回结果即可。如采用递归遍历，结果可设为一个全局变量ans，如需辅助变量也可设为全局变量，否则应采用传引用的方式传入遍历函数。
以上题目也个改为，前序、后序遍历，或寻找前序节点，或寻找节点最大值
节点值为偶数的个数等。

递归解法

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode *ans,*pre;
    TreeNode* inorderSuccessor(TreeNode* root, TreeNode* p) {
        ans=pre=NULL;
        dfs(root,p);
        return ans;
    }
    void dfs(TreeNode* root,TreeNode* p){
        if(root==NULL) return;
        dfs(root->left,p);
        if(pre==p) ans=root;
        pre=root;
        dfs(root->right,p);
    }
};

同样，如果寻找前序节点，只需将条件替换为：

if(root==p) ans=pre;
pre=root;

条件复杂一点，寻找最大节点的后续节点：

TreeNode *ans,*pre,*maxnode;

if(maxnode==NULL || root->val>maxnode->val)
    maxnode=root;
else if(pre==maxnode)
    ans=root;
pre=root;

在树中寻找节点了问题，几乎都可以用此类方法解决。递归解法思路清晰，容易写，但同样也可以写为非递归的形式。

class Solution {
public:
    TreeNode* inorderSuccessor(TreeNode* root, TreeNode* p) {
        TreeNode* cur=root;
        TreeNode* pre=NULL;
        stack<TreeNode*> stk;
        while(cur || !stk.empty()){
            while(cur){
                stk.push(cur);
                cur=cur->left;
            }
            cur=stk.top();
            stk.pop();
            if(pre==p) return cur;
            pre=cur;
            cur=cur->right;
        }
        return NULL;
    }
};

树遍历的非递归写法可参考 二叉树的非递归遍历