程序员人生 网站导航

Binary Tree Inorder Traversal -- leetcode

栏目:php教程时间:2015-05-07 09:28:52

Given a binary tree, return the inorder traversal of its nodes' values.

For example:
Given binary tree {1,#,2,3},

1 2 / 3

return [1,3,2].

Note: Recursive solution is trivial, could you do it iteratively?


实现中序遍历,且不能用递归。

算法1:使用栈

中序遍历为,先访问左子树,再访问根。

访问完左子树,需要要回到根。而树本身并没有存储到根的指针。

故需要栈的帮助,存储指向父结点的指针。即先把自己入栈,再进入左子树。

此代码在leetcode上,实际履行时间为5ms。

/** * Definition for binary tree * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */ class Solution { public: vector<int> inorderTraversal(TreeNode *root) { vector<int> ans; stack<TreeNode *> s; while (!s.empty() || root) { while (root) { s.push(root); root = root->left; } root = s.top(); s.pop(); ans.push_back(root->val); root = root->right; } return ans; } };


以上代码,也能够写作:

vector<int> inorderTraversal(TreeNode *root) { vector<int> ans; stack<TreeNode *> s; while (!s.empty() || root) { if (root) { s.push(root); root = root->left; } else { root = s.top(); s.pop(); ans.push_back(root->val); root = root->right; } } return ans; }


算法2  Morris Traversal

利用空闲的right指针指向根结点。这样在访问完左子树时,根据右孩子,回到根结点。

中序遍历,要求,先访问左子树的所有结点,然后访问该结点。

那末根结点root的左子树中最右下的结点rightMost必为左子树最后1个被访问的结点。而此结点rightMost的右指针,必为空。  则可以复用此右指针,指向根结点root。 这样,在访问完rightMost后,就能够回到根结点root了。

故,在进入左子树之前,先找到左子树中最右下的结点,让其right指针,指向自己。未思进,先思退。留好退路。

这样,rigth指针就有2义性,即表示右子树,又表示先人结点。如何避免2义性呢,同时也为了不死循环。

那就是,在寻觅其左子树的最右下结点时,如果又回到了自己。说明它左子树,已全部访问完成。

即,寻觅最右结点时,会出现两种情况,1种是碰到right为空;1种是回到了自己。

此代码在leetcode上实际履行时间为4ms。

vector<int> inorderTraversal(TreeNode *root) { vector<int> ans; while (root) { if (!root->left) { ans.push_back(root->val); root = root->right; } else { TreeNode *rightMost = root->left; while (rightMost->right && rightMost->right != root) rightMost = rightMost->right; if (!rightMost->right) { rightMost->right = root; root = root->left; } else { ans.push_back(root->val); rightMost->right = 0; root = root->right; } } } return ans; }



------分隔线----------------------------
------分隔线----------------------------

最新技术推荐