用递归和非递归的方法实现二叉树遍历。
递归中序遍历:
struct TreeNode
{
int val;
TreeNode* left;
TreeNode* right;
};
void InOrderRec(TreeNode* root)
{
if (root == NULL)
return;
if (root->left)
InOrderRec(root->left);
cout << root->val;
if (root->right)
InOrderRec(root->right);
}
非递归中序遍历:
使用栈来保存中间节点。
1.先将根节点压栈。
2.将左子节点压栈,直至左子节点不存在,此时位于栈顶的应该是最左侧的根节点。
3.将栈顶的结点打印并弹出。
4.将当前节点设为右子节点,返回1循环。
void InOrderUnRec(TreeNode * root)
{
if (root == NULL)
return;
stack<TreeNode*> s;
TreeNode* node = root;
while (node != NULL || !s.empty())
{
while (node != NULL)
{
s.push(node);
node = node->left;
}
if (!s.empty())
{
node = s.top();
s.pop();
cout << node->val << endl;
node = node->right;
}
}
}
递归方法实现另外两种遍历相对简单,这里就不赘述。着重介绍非递归方法。
非递归方法实现前序遍历:
使用栈保存中间节点。对于任一节点node
1.将node输出,然后压栈。
2.判断node的左子树存在,若不存在,则取出栈顶元素并出栈,同时将栈顶元素的右子节点设为node,压栈,执行1的操作。若不为空,则将左子树赋值为node。
3.直到node为空并且栈为空,遍历结束。
void InOrderUnRec(TreeNode * root)
{
if (root == NULL)
return;
stack<TreeNode*> s;
TreeNode* node = root;
while (node != NULL || !s.empty())
{
while (node != NULL)
{
cout<<node->val<<endl;
s.push(node);
node = node->left;
}
if (!s.empty())
{
node = s.top();
s.pop();
node = node->right;
}
}
}
非递归方法实现后序遍历:
后序遍历相对于前面两个比较难。对任意节点,要先访问其左子节点再访问其右子节点。因此,要设置一个标志位,表示右子节点是否被访问过。
对任意一个节点node,初始化flag=false,表示第一次访问,即右子节点还没有访问过。
1.将node压入栈。
2.判断左子节点是否存在:如果存在,则将node赋值为左子节点,循环至1的操作;若不存在,则判断node的flag是否为false,是false,将node赋值为右子节点同时将flag赋值为true,循环至1的操作,若是true,表示右子节点已经遍历过,则输出node,将栈顶元素弹出,node赋值为null。
3.直至node为空并且栈为空,遍历结束。
struct TreeNodeWithFlag
{
int val;
TreeNodeWithFlag* left;
TreeNodeWithFlag* right;
bool visited;
};
void PostOrderUnRec(TreeNodeWithFlag* root)
{
if (root == NULL)
return;
TreeNodeWithFlag* node = root;
stack<TreeNodeWithFlag*> s;
while (node != NULL || !s.empty())
{
while (node)
{
s.push(node);
node = node->left;
}
if (!s.empty())
{
node = s.top();
if (node->visited == false)
{
node->visited = true;
node = node->right;
}
else
{
cout << node->val << " ";
s.pop();
node = NULL;
}
}
}
}