【剑指offer】按之字形顺序打印二叉树（树）

本文介绍了一种特殊的二叉树遍历方法——之字形打印。此算法按照从左到右，再从右至左的顺序交替打印二叉树的每一层节点，实现了对二叉树的美观展示。

题目描述

请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。

链接

https://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0?tpId=13&tqId=11212&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

代码

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


class Solution {
public:
    vector<vector<int> > Print(TreeNode* pRoot) {
        vector<vector<int> > ans;
        vector<TreeNode* > father;
        vector<TreeNode* > child;
        if(pRoot == NULL){
            return ans;
        }
        int flag = 0; // 0 表示从左到右
        father.push_back(pRoot);
        while(father.size()>0){
            vector<int> temp;
            if(flag == 0){
                for(int i = 0; i < father.size(); i++){
                    temp.push_back(father[i]->val);
                }
                flag = 1;
                ans.push_back(temp);
            }
            else{
                for(int i = int(father.size() - 1); i >= 0; i--){
                    temp.push_back(father[i]->val);
                }
                flag = 0;
                ans.push_back(temp);
            }
            // 添加下一层，按照从左到右添加
            for(int i = 0; i < father.size(); i++){
                if(father[i]->left){
                    child.push_back(father[i]->left);
                }
                if(father[i]->right){
                    child.push_back(father[i]->right);
                }
            }
            father = child;
            child.clear();
        }
        return ans;
    }
};