Morris先序遍歷實現空間複雜度為O(1)
阿新 • • 發佈:2019-02-13
先序的遍歷這裡的實現是根,左,右。
從根出發
如果其左孩子不為空
從左孩子出發,一直往右邊走(迭代獲取其右孩子),如果其右孩子能走到尾部,即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;
}