方法一:递归遍历,记录全局变量的父节点,当parent == target的时候,当前节点就是答案。
/**
* 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* res;
TreeNode* parent;
void dfs(TreeNode* root,TreeNode* p){
if(!root)
return ;
dfs(root->left, p);
if(!res&&parent==p){
res=root;
return ;
}
parent=root;
dfs(root->right, p);
}
TreeNode* inorderSuccessor(TreeNode* root, TreeNode* p) {
res=nullptr;
parent=nullptr;
dfs(root, p);
return res;
}
};
方法二:迭代
/**
* 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) {
TreeNode* parent=nullptr, *node;
stack<TreeNode*> stk;
node=root;
while(stk.size()||node){
while(node){
stk.push(node);
node=node->left;
}
node=stk.top();
stk.pop();
if(parent == p)
return node;
parent = node;
node=node->right;
}
return nullptr;
}
};