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;

}

点赞