二叉树:前序/中序/后序/层序遍历 (递归&非递归 c++版本)

参考 https://www.cnblogs.com/bigsai/p/11393609.html

1. 前序遍历

前序的规则就是根结点 ---> 左子树 ---> 右子树.我们在调用递归前进行节点操作。对于前序,就是先访问(输出)该节点。而递归左,递归右侧,会优先递归左侧。直到没有左节点。才会停止。访问次序大致为:

测试用例:


(1) 递归版本

struct TreeNode {
   int val;
   TreeNode *left;
   TreeNode *right;
   TreeNode(int x): val(x), left(NULL), right(NULL) {}
};
void preOrder(TreeNode *root) {
    if (root) {
        std::cout << root->val << ",";
        preOrder(root->left);
        preOrder(root->right);
    }
}
// 以下为测试例子
int main() {
    TreeNode *root1 = new TreeNode(3);
    TreeNode *left1 = new TreeNode(9);
    TreeNode *right1 = new TreeNode(20);
    TreeNode *left2 = new TreeNode(15);
    TreeNode *right2 = new TreeNode(7);
    root1 -> left = left1;
    root1 -> right = right1;
    right1 -> left = left2;
    right1 -> right = right2;
    preOrder(root1);
}

前序遍历结果:3,9,20,15,7

(2) 非递归版本

利用栈,栈的顺序为先进先出。那么顺序如何添加?递归是左递归,右递归。但是利用栈要相反,因为如果左进栈、右进栈会出现以下后果:

所以,利用递归的思路,需要先放右节点进栈,再放左节点进栈,这个下次·再取节点取到左节点·,这个节点再右节点进栈,左节点进栈。然后循环一直到最后会一直优先取到左节点。达到和递归顺序相仿效果。

void preOrder(TreeNode *root) {
    stack<TreeNode *> s;
    s.push(root);
    while (!s.empty()) {
        TreeNode *curr = s.top();
        std::cout << curr->val << ",";
        s.pop();
        if (curr -> right)
            s.push(curr -> right);
        if (curr -> left)
            s.push(curr -> left);
    }
}

2. 中序遍历

中序遍历的规则是:左子树---> 根结点 ---> 右子树。所以我们访问节点的顺序需要变

(1) 递归版本

struct TreeNode {
   int val;
   TreeNode *left;
   TreeNode *right;
   TreeNode(int x): val(x), left(NULL), right(NULL) {}
};

void midOrder(TreeNode *root) {
    if (root) {
        midOrder(root->left);
        std::cout << root->val << ",";
        midOrder(root->right);
    }
}
int main() {
    TreeNode *root1 = new TreeNode(3);
    TreeNode *left1 = new TreeNode(9);
    TreeNode *right1 = new TreeNode(20);
    TreeNode *left2 = new TreeNode(15);
    TreeNode *right2 = new TreeNode(7);
    root1 -> left = left1;
    root1 -> right = right1;
    right1 -> left = left2;
    right1 -> right = right2;
    midOrder(root1);
}

中序遍历结果:9,3,15,20,7

(2) 非递归版本

中序排列的顺序是: 左节点,根节点,右节点。那么我们在经过根节点的前面节点不能释放, 因为后面还需要用到它。所以要用栈先储存。
它的规则大致为:

  • 栈依次存入左节点所有点,直到最左侧在栈顶。
  • 开始抛出栈顶并访问。(例如第一个抛出2)。如果有右节点。那么将右节点加入栈中,然后右节点一致左下遍历直到尾部。


void midOrder2(TreeNode *root) {
    stack<TreeNode *> s;
    TreeNode *p = root;
    while (!s.empty() || p) {
       //一直遍历到左子树最下边,边遍历边保存根节点到栈中
       while(p) {
           s.push(p);
           p = p->left;
       }
       // 当p为空时,说明已经到达左子树最下边,这时需要出栈了
       if (!s.empty()) {
         p = s.top();
         std::cout << p->val << ",";
         s.pop();
         p = p->right;
       }
    }
}

3. 后序遍历

后序遍历就是左子树 ---> 右子树 ---> 根结点

(1) 递归版本

struct TreeNode {
   int val;
   TreeNode *left;
   TreeNode *right;
   TreeNode(int x): val(x), left(NULL), right(NULL) {}
};

void postOrder(TreeNode *root) {
    if (root) {
        postOrder(root->left);
        postOrder(root->right);
        std::cout << root->val << ",";
    }
}

int main() {
    TreeNode *root1 = new TreeNode(3);
    TreeNode *left1 = new TreeNode(9);
    TreeNode *right1 = new TreeNode(20);
    TreeNode *left2 = new TreeNode(15);
    TreeNode *right2 = new TreeNode(7);
    root1 -> left = left1;
    root1 -> right = right1;
    right1 -> left = left2;
    right1 -> right = right2;
    postOrder(root1);
}

后序遍历结果:9,15,7,20,3

(2) 非递归版本

void postOrder2(TreeNode *root) {
    stack<pair<TreeNode*, bool> > s;
    TreeNode *p = root;
    while (!s.empty() || p) {
            while (p) {
                s.push(make_pair(p, false));
                p = p->left;
            }
            if (!s.empty()) {
                pair<TreeNode*, bool> curr = s.top();
                if (curr.second) {
                     std::cout << curr.first->val <<  ",";
                     s.pop();
                }
                else {
                    curr.second = true;
                    p = p->right;
                }
            }
    }
}

4. 层序遍历

https://blog.csdn.net/FX677588/article/details/74276513
层序遍历所要解决的问题很好理解,就是按二叉树从上到下,从左到右依次打印每个节点中存储的数据。如下图:


按层序遍历的原则,打印顺序依次应该是:A->B->C->D->E->F->G

void levelOrder(TreeNode *root){
    queue<TreeNode *> fatherQueue, childQueue;
    TreeNode *p = root;
    if (p)
        fatherQueue.push(p);
    while (!fatherQueue.empty()) {
        p = fatherQueue.front();
        std::cout << p->val << ",";
        fatherQueue.pop();
        if (p->left)
            childQueue.push(p->left);
        if (p->right)
            childQueue.push(p->right);
        if (fatherQueue.empty() && !childQueue.empty()) {
            swap(fatherQueue, childQueue);
        }
    }
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,919评论 6 502
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,567评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 163,316评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,294评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,318评论 6 390
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,245评论 1 299
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,120评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,964评论 0 275
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,376评论 1 313
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,592评论 2 333
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,764评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,460评论 5 344
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,070评论 3 327
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,697评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,846评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,819评论 2 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,665评论 2 354

推荐阅读更多精彩内容