力扣日常打卡---面试题04.06.后继者

题目描述：
设计一个算法，找出二叉搜索树中指定节点的“下一个”节点（也即中序后继）。
如果指定节点没有对应的“下一个”节点，则返回null。
示例：

思路及题解：
由于题目交代输入树已为二叉搜索树，那么，搜寻后继者的思路就变得清晰。
思路1：用一个vector储存中序遍历的结果，然后找到p所对应的值，则后继者为数组中p的值的后一个值，若p的值为数组中最后一个值，则返回null。代码如下：

vector<TreeNode*> temp;
void helper(TreeNode* node)
{
	if(!node)return nullptr;
	helper(node->left);
	temp.push_back(node);
	helper(node->right);
}
/*省略中间代码，只贴具体实现代码*/
for(int i=0;i<temp.size();i++)
{
	if(i==temp.size()-1)return nullptr;
	else if(temp[i]->val==p->val)
	{
		return temp[i+1];
	}
}

从上述代码可知：空间复杂度为O(n),时间复杂度为O(n)，其中n为树的节点数，没有充分运用二叉搜索树的性质。

那么，怎么样利用二叉搜索树的性质呢？我们知道，二叉搜索树的左节点值<节点值<右节点值，切无重复，那么我们将p节点的值与当前节点的值进行比较，若p节点的值与当前节点的值相等或者大于当前节点的值，那么必定，p节点的值一定存在于当前节点的右子树中！，即：

if(p->val>=node->val)
	return inorderSuccessor(node->right,p);

反之，若p节点的值小于当前节点的值，p点的后继者则要分为两种情况，**一种是，当前节点本身，另一种是存在于左子树中。**那么，怎么区分这两种情况呢，很简单，对于当前节点而言，若递归返回的值为nullptr，说明在当前节点的左子树中没有找到p的后继者，又p点的值小于当前节点的值，那么，**只能是p的后继者即为当前节点！**反之，若递归返回的值不为nullptr，则返回找到的节点。反映在代码上为：

if(p->val<node->val)
{
	TreeNode* nxt = inorderSuccessor(node->left,p);
	return nxt==nullptr ? node : nxt;
}

综合上述两种情况，可得出其时间复杂度为O(logn),空间复杂度为O(1)。 大大减少了其时间与空间的开销。

最终代码及运行结果如下:

/**
 * 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* inorderSuccessor(TreeNode* root, TreeNode* p) {
        if(!root||!p)return NULL;
        int num  = p->val;
        int cmp = root->val;
        if(num>=cmp)return inorderSuccessor(root->right,p);
        else 
        {
            TreeNode* nxt =  inorderSuccessor(root->left,p);
            return nxt ? nxt:root;
        }
    }
};

其实还有更多的方法，具体详见力扣题解：
后继者-五大解法汇总