用两种方法先序遍历:
1、访问根节点—根节点入栈—若有左孩子,访问,入栈;若无,回到父节点(top)找右孩子访问右孩子;循环入栈。
2、循环 :每次访问根节点—右孩子压栈—访问左孩子。
原理:先序遍历:根左右,所以访问根—把右孩子压栈—先访问左孩子。
代码如下:
#include <iostream>
#include "stack"
using namespace std;
struct TNode
{
TNode *left;
TNode *right;
int data;
};
TNode *Root;
void Visit(TNode *cur)
{
cout << cur->data << endl;
}
void preOrder1(TNode* root)
{
stack<TNode*> S;
while (( root != NULL) || !S.empty())
{
if (root != NULL)
{
// 获取根节点的值
Visit(root);
// 压入栈中
S.push(root);
root = root->left; // 依次访问左子树
}
else
{
root = S.top();
S.pop();
root = root->right;
}
}
}
stack<TNode*> ST2;
void preOrder2(TNode *root) //Method 2
{
ST2.push(root);
while (!ST2.empty())
{
if( root != NULL )
{
Visit(root);
if(root->right!=NULL)
ST2.push(root->right);
root = root->left;
}
else
{
root=ST2.top();
ST2.pop();
}
}
}
void InitTree(TNode *root)
{
root->data = 1;
TNode* tree[10];
for (int i=0; i<7; i++)
{
tree[i] = (TNode*) malloc(sizeof(TNode*));
tree[i]->data = i;
tree[i]->left = tree[i]->right = NULL;
}
tree[4]->right=tree[6];
tree[1]->left=tree[3];
tree[1]->right=tree[4];
tree[2]->left=tree[5];
tree[0]->left=tree[1];
tree[0]->right=tree[2];
Root = tree[0];
}
void main()
{
Root = (TNode*)malloc(sizeof(TNode*));
InitTree(Root);
// preOrder1(Root);
preOrder2(Root);
}