1. 程式人生 > >Morris先序遍歷實現空間複雜度為O(1)

Morris先序遍歷實現空間複雜度為O(1)

先序的遍歷這裡的實現是根,左,右。

從根出發

如果其左孩子不為空

           從左孩子出發,一直往右邊走(迭代獲取其右孩子),如果其右孩子能走到尾部,即n=n->right經過若干次之後n為空

           說明當前孩子沒有訪問過,否則認為當前孩子已經被訪問過,則把當前節點設定為當前節點的右孩子。

如果左孩子為空

            則返回到當前節點的上一個節點

vector<int> preorderTraversal(TreeNode *root) {

vector<int> result;
TreeNode *cur = root;
while (cur != nullptr) {

if (cur->left == nullptr) {//對於內部節點,這個地方的程式碼只是普通的意思,左孩子為空把右孩子放進去

                                                        //對於葉子節點來說是獲取其前一個節點

result.push_back(cur->val);
cur = cur->right;
}
else {

TreeNode *node = cur->left;
while (node->right != nullptr && node->right != cur)

node = node->right;

                       這個判斷主要是判斷當前節點有沒有被訪問過

if (node->right == nullptr) { 
result.push_back(cur->val); 
node->right = cur;
cur = cur->left;
}
else { 
node->right = nullptr;
/* prev = cur; ̼*/
cur = cur->right;
}
}
}
return result;
}