数据结构-前序遍历、中序遍历、后序遍历、层级遍历(递归、非递归)

短命女 2022-07-24 05:01 320阅读 0赞

二叉树的遍历是一个非常基础又重要的内容。遍历就是访问二叉树中的每一个节点,并且每个节点只访问一次。二叉树的遍历分为前序遍历、中序遍历、后序遍历和层级遍历。

前序遍历

前序遍历的顺序是:根节点 -> 左子节点 -> 右子节点。

这里写图片描述

上图二叉树的前序遍历结果是: ABCDEFGH

递归实现

按照前序遍历的顺序,可以非常快写出其递归实现

  1. void PreOrderTraverseRec (Node* root) {
  2. if (root != NULL) {
  3. printf("%d ", root->val);
  4. PreOrderTraverseRec (root->left);
  5. PreOrderTraverseRec (root->right);
  6. }
  7. }

非递归实现

前序遍历的非递归实现可以使用栈模拟。

版本1

其实在前序遍历的内部实现,每次遍历一个节点后,接下来先遍历这个节点的左子节点,对这个左子节点将其看成根节点,又递归遍历其左子节点,直到访问到叶节点。将叶节点弹栈,得到其父节点。因为已经访问过父节点,以及父节点的左子节点,所以接下来应该按照上述方法递归访问父节点的右子树。

  1. void PreOrderTraverseNonRec (Node* root) {
  2. if (root == NULL) {
  3. printf("empty tree!\n");
  4. return;
  5. } else {
  6. //method 1
  7. stack<Node*> s;
  8. Node* p = root;
  9. while ( p != NULL || !s.empty()) {
  10. while (p != NULL) {
  11. printf("%d ", p->val);
  12. s.push (p);
  13. p = p->left;
  14. }
  15. if (!s.empty()) {
  16. p = s.top ();
  17. s.pop ();
  18. p = p->right;
  19. }
  20. }
  21. }
  22. }

版本2

还有另外一种更加简洁的用栈模拟的方法。前序遍历的顺序是遍历根节点 -> 左子节点 -> 右子节点。根据栈的后进先出的特点,可以按照根节点 -> 右节点 -> 左子节点的顺序将节点压入栈中。这样每次压栈之前访问根节点,在弹栈先访问左子节点,再到右子节点。

  1. void PreOrderTraverseNonRec (Node* root) {
  2. stack<Node*> s;
  3. Node* cur = NULL;
  4. s.push (root);
  5. while (!s.empty()) {
  6. cur = s.top ();
  7. printf("%d ", cur->val);
  8. s.pop (cur);
  9. if (cur->right)
  10. s.push (cur->right);
  11. if (cur->left)
  12. s.push (cur->left);
  13. }
  14. }
  15. }

中序遍历

中序遍历的顺序是: 根节点的左子节点 -> 根节点 -> 根节点的右子节点

递归实现

  1. void InOrderTraverseRec (Node* root) {
  2. if (root != NULL) {
  3. InOrderTraverseRec (root->left);
  4. printf("%d ", root->val);
  5. InOrderTraverseRec (root->right);
  6. }
  7. }

非递归实现

这里写图片描述

中序遍历的非递归实现与前序遍历非递归实现的版本1很相似,只不过输出根节点的先后顺序不同。

  1. void InOrderTraverseNonRec (Node* root) {
  2. if (root == NULL) {
  3. printf("empty tree!\n");
  4. return;
  5. } else {
  6. stack<Node*> s;
  7. Node* p = root;
  8. while (p != NULL || !s.empty()) {
  9. while (p != NULL) {
  10. s.push (p);
  11. p = p->left;
  12. }
  13. if (!s.empty()) {
  14. p = s.top ();
  15. s.pop ();
  16. printf("%d ", p->val);
  17. p = p->right;
  18. }
  19. }
  20. }
  21. }

后序遍历

后续遍历的顺序为: 左子节点 -> 右子节点 -> 根节点

递归实现

  1. void PostOrderTraverseRec (Node* root) {
  2. if (root != NULL) {
  3. PostOrderTraverseRec (root->left);
  4. PostOrderTraverseRec (root->right);
  5. printf("%d ", root->val);
  6. }
  7. }

非递归实现

后序遍历

版本1

由于后续遍历中需要先访问左、右子节点,再来访问根节点,因此非递归实现应该为:访问根节点,再向左不断访问左子节点,直到访问到叶子节点,这时不能将这个叶子节点弹栈,因为其右子节点还没有被访问。所以,应该按照上述方法访问完叶节点的右子树,再来访问叶节点。为了区别出两次访问顺序,可以使用一个变量记录根节点是否已经被访问过。

  1. void PostOrderTraverseNonRec2 (Node* root)
  2. {
  3. if (root == NULL) return;
  4. stack <Node*> s;
  5. Node *p = root;
  6. while (p != NULL || !s.empty()) {
  7. while (p != NULL) {
  8. p->isFirstVisited = true;
  9. s.push(p);
  10. p = p->left;
  11. }
  12. while ( !s.empty()) {
  13. p = s.top();
  14. if (p->isFirstVisited) {
  15. p->isFirstVisited = false;
  16. p = p->right;
  17. } else {
  18. printf("%d ", p->val);
  19. s.pop();
  20. p = NULL;
  21. }
  22. }
  23. }
  24. }

版本2

第一种方法需要额外的空间存储节点是否被访问过,浪费空间。可以直接使用一个指针记录当前访问的节点的前一个被访问的节点,如果前一个被访问的节点正好是当前节点的右子节点,则说明右子节点已经被访问过,直接输出当前节点即可。

  1. void PostOrderTraverseNonRec (Node* root) {
  2. stack<Node*> s;
  3. Node *cur = root, *pre = NULL;
  4. while (cur != NULL || !s.empty()) {
  5. while (cur != NULL) {
  6. s.push (cur);
  7. cur = cur->left;
  8. }
  9. if (!s.empty()) {
  10. cur = s.top ();
  11. if (cur->right == NULL || cur->right == pre) {
  12. printf("%d ", cur->val);
  13. pre = cur;
  14. s.pop ();
  15. cur == NULL;
  16. } else {
  17. cur = cur->right;
  18. }
  19. }
  20. }
  21. }

层级遍历

层级遍历

层级遍历就是按照层次的顺序,从上往下一层一层遍历二叉树。

递归实现

为了能够逐层遍历二叉树,必须知道二叉树的高度,然后才能逐层遍历。而在每一层中,先遍历根节点的左子节点,再遍历右子节点。

  1. int GetHeight (Node* root) {
  2. if(root == NULL)
  3. return 0;
  4. else
  5. return max(GetHeight(root->left) + 1, GetHeight(root->right) + 1);
  6. }
  7. void PrintLevel (Node* root, int level) {
  8. if (root == NULL)
  9. return;
  10. if (level == 1) {
  11. printf("%d ", root->val);
  12. return;
  13. }
  14. return ( PrintLevel(root->left, level-1) + PrintLevel( root->right, level-1));
  15. }
  16. void LevelTraverseRec (Node* root) {
  17. int height = GetHeight (root);
  18. for (int i = 1; i <= height; ++i)
  19. {
  20. PrintLevel (root, i);
  21. }
  22. }

非递归实现

层级遍历可以使用队列来模拟:每次访问根节点R,再分别访问根节点的左右子节点L、R,再继续访问L、R的左右子节点。这就符合队列的先进先出的特定。

  1. void LevelTraverseNonRec (Node* root) {
  2. if (root == NULL) return;
  3. Node* cur = root;
  4. queue<Node*> q;
  5. q.push (root);
  6. while (!q.empty()) {
  7. cur = q.front ();
  8. printf("%d ", cur->val);
  9. q.pop ();
  10. if (cur->left)
  11. p.push (p->left);
  12. if (cur->right)
  13. p.push (p->right);
  14. }
  15. }

发表评论

表情:
评论列表 (有 0 条评论,320人围观)

还没有评论,来说两句吧...

相关阅读