897. 遞增順序搜索樹

Ideas

看到搜索二叉樹就想到了它的中序遍歷序列是有序的，所以干脆直接用中序遍歷序列，把每一項的left結點都置為nullptr，right結點置為下一項就OK了。

需要注意中序遍歷序列的最后一個節點需要做一些調整：left和right指針都置為空。不然可能會導致出現環。

Code

C++

class Solution { public:vector<TreeNode*> middle_order_traversal(TreeNode* node) {TreeNode* item = node;stack<TreeNode*> stk;vector<TreeNode*> ans;while (!stk.empty() || item != nullptr) {if (item != nullptr) {stk.push(item);item = item->left;} else {item = stk.top();stk.pop();ans.push_back(item);item = item->right;}}return ans;} TreeNode* increasingBST(TreeNode* root) {vector<TreeNode*> values = middle_order_traversal(root);for (int i = 0; i < values.size(); i++) {if (i == values.size() - 1) {values[i]->left = nullptr;values[i]->right = nullptr;} else {cout << values[i]->val << ", ";values[i]->left = nullptr;values[i]->right = values[i + 1];}}return values[0];} };

總結

以上是生活随笔為你收集整理的LeetCode Algorithm 897. 递增顺序搜索树的全部內容，希望文章能夠幫你解決所遇到的問題。

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