用递归和非递归的方法实现二叉树遍历

来源：互联网发布：linux下c ide 编辑：程序博客网时间：2024/06/05 00:29

用递归和非递归的方法实现二叉树遍历。

递归中序遍历：

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;            }        }    }}

0 0