Binary Tree Traversal

时间:2023-03-08 16:37:25

1、Preorder Traversal

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

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

   1
\
2
/
3

return [1,2,3].

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

解题:前序遍历首先访问根结点然后遍历左子树,最后遍历右子树。在遍历左、右子树时,仍然先访问根结点,然后遍历左子树,最后遍历右子树。这里依然使用栈实现。先访问根结点,再将右结点、左结点压到栈中,根据栈先进后出的特性,会先遍历左子树,再遍历右子树。

/**
* 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> preorderTraversal(TreeNode *root) {
vector<int> ans;
stack<TreeNode*> que;
TreeNode *tmp;
if(root != NULL){ que.push(root);
while(!que.empty())
{
tmp = que.top();
que.pop();
ans.push_back(tmp->val); if(tmp->right != NULL)
que.push(tmp->right);
if(tmp->left != NULL)
que.push(tmp->left);
}
}
return ans;
}
};

 2、 Inorder Traversal

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

   1
\
2
/
3

return [1,3,2].

解题:中序遍历首先遍历左子树,然后访问根结点,最后遍历右子树。在遍历左、右子树时,仍然先遍历左子树,然后访问根结点,最后遍历右子树。这里依然使用栈实现。

/**
* 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*> stk;
TreeNode* tmp = root;
while(tmp != NULL || !stk.empty())
{
while(tmp != NULL)
{
stk.push(tmp);
tmp = tmp->left;
}
if(!stk.empty())
{
tmp = stk.top();
ans.push_back(tmp->val);
stk.pop();
tmp = tmp->right;
}
}
return ans;
}
};

3、Postorder Traversal

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

   1
\
2
/
3

return [3,2,1].

解题:后序遍历首先遍历左子树,然后遍历右子树,最后访问根结点。算法处理过程:先迭代并压栈至左子树最左结点,取出栈顶元素,判断是否具有右儿子(或者其右儿子刚刚被访问),如果条件为真,便访问该结点,否则对其右儿子进行后序遍历。

/**
* 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> postorderTraversal(TreeNode *root) {
vector<int> ans;
stack<TreeNode*> stk;
TreeNode *pre,*tmp = root;
while(tmp != NULL || !stk.empty()){
while(tmp != NULL) {
stk.push(tmp);
tmp = tmp->left;
}
tmp = stk.top();
if(tmp->right == NULL || tmp->right == pre){
ans.push_back(tmp->val);
stk.pop();
pre = tmp;
tmp = NULL;
}
else {
tmp = tmp->right;
}
}
return ans;
}
};