1. 題目

設計一個算法，找出二叉搜索樹中指定節點的“下一個”節點（也即中序后繼）。

如果指定節點沒有對應的“下一個”節點，則返回null。

示例 1: 輸入: root = [2,1,3], p = 12/ \ 1 3輸出: 2示例 2: 輸入: root = [5,3,6,2,4,null,null,1], p = 65/ \3 6/ \2 4/ 1輸出: null

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/successor-lcci
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

• 循環寫法，中序遍歷即可

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

總結

以上是生活随笔為你收集整理的程序员面试金典 - 面试题 04.06. 后继者（循环中序遍历）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。