参考博客:二叉树的递归和非递归方式的三种遍历
1、前序遍历
/*
先遍历根节点及其左子树,输出,对遍历到的每一个父节点压栈,然后访问栈顶节点的右子树
对应于先序遍历的先遍历父节点,再遍历左子树再遍历右子树的顺序
*/
void preOrderNoRecursion(treeNode* root) {
if (!root) {
cout << "empty tree!\n";
return;
}
stack<treeNode*> s;
while (!s.empty() || root) {
if (root) {
cout << root->value << " ";
s.push(root);
root = root->left;
} else {
root = s.top();
s.pop();
root = root->right;
}
}
}
2、中序遍历
/*
先遍历根节点及其左子树,对遍历到的每一个父节点压栈,然后输出,再访问栈顶节点的右子树
与先序遍历类似,只是打印数据时机不同
*/
void inOrderNoRecursion(treeNode* root) {
if (!root) {
cout << "empty tree!\n";
return;
}
stack<treeNode*> s;
while (!s.empty() || root) {
if (root) {
s.push(root);
root = root->left;
} else {
root = s.top();
s.pop();
cout << root->value << " ";
root = root->right;
}
}
}
3、后序遍历
/*
由两个栈实现,
先把根节点压入第一个栈,
当栈1不为空时,
出栈,并压入第二个栈,然后将刚才出栈的栈顶元素的左右孩子入栈1;
当栈1空了,打印出栈2内的元素
栈1的栈顶优先为靠上层的右孩子再到下层的左孩子
栈2用于保存栈1的栈顶
入栈后符合后序遍历 上层右孩子后输出,下层左孩子先输出的顺序
*/
void postOrderNoRecursion(treeNode* root) {
if (!root) {
cout << "empty tree!\n";
return;
}
stack<treeNode*> s1, s2;
s1.push(root);
while (!s1.empty()) {
treeNode* current = s1.top();
s1.pop();
s2.push(current);
if (current->left) s1.push(current->left);
if (current->right) s1.push(current->right);
}
while (!s2.empty()) {
treeNode* top = s2.top();
s2.pop();
cout << top->value << " ";
}
}