先序遍历
void preOrderS(Node* root)
{
stack<Node*> s;
Node* p = root;
while (p || !s.empty())
{
while (p)
{
cout << p->data;
s.push(p);
p = p->left;
}
if (!s.empty())
{
p = s.top();
s.pop();
p = p->right;
}
}
}
思路:令p=root,不断将p=p->left,并打印节点,也就是不停的取左节点直到为空。如果发现p为空,那么说明已经到了最左节点,出栈之后p=p->right,继续循环判断。
中序遍历
void inOrderS(Node* root)
{
stack<Node*> s;
Node* p = root;
while (p || !s.empty())
{
while (p)
{
s.push(p);
p = p->left;
}
if (!s.empty())
{
p = s.top();
s.pop();
cout << p->data;
p = p->right;
}
}
}
思路:和前序遍历非常像。令p=root,不断将p=p->left,也就是不停的取左节点直到为空。如果发现p为空,那么就说明已经到了最左节点,出栈并打印当前节点,之后p=p->right,继续循环判断。
后序遍历
void preOrderS(Node* root)
{
stack<Node*> s1;
stack<Node*> s2;
Node* p = root;
s1.push(p);
while (!s1.empty())
{
p = s1.top();
s1.pop();
s2.push(p);
if (p->left)s1.push(p->left);
if (p->right)s1.push(p->right);
}
while (!s2.empty())
{
cout << s2.top();
s2.pop();
}
}
思路:用两个栈,先将头结点入栈s1。之后将该节点出栈s1,入栈s2,同时将该节点的左孩子入栈,右孩子入栈。依次判断s1栈顶中节点是否有左右孩子,如果没有则入栈s2,往复循环。最终依次出s2栈即为后序遍历。