剑指Offer-按之字形顺序打印二叉树

题目描述

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

整体思路

看到这个题目的第一个想法是和按行打印二叉树很相似，而按行打印二叉树是利用队列来实现的；但是这里要求按之字形，于是考虑用两个栈来实现。两个栈分别存放奇数行的节点和偶数行的节点，考虑到奇数行从左到右打印，偶数行从右到左打印；

1）初始的时候栈stack1存放根节点，栈stack2为空；

2）遍历栈stack1的节点的时候，并按先左节点后又右节点的顺序，将栈stack1中节点的子节点加入栈stack2，直至stack1为空

3）遍历栈stack1的节点的时候，并按先右节点后又左节点的顺序，将栈stack2中节点的子节点加入栈stack1，直至stack2为空

重复2）3）直至栈stack1和stack2均为空

代码实现

/*
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> > res;
        if(pRoot == NULL)
            return res;
        stack<TreeNode*> stack1, stack2;
        stack1.push(pRoot);
        while(!stack1.empty() || !stack2.empty()){
            vector<int> tmv;
            TreeNode* tmp;
            while(!stack1.empty()){
                tmp = stack1.top();
                tmv.push_back(tmp->val);
                if(tmp->left)
                    stack2.push(tmp->left);
                if(tmp->right)
                    stack2.push(tmp->right);
                stack1.pop();
            }
            if(tmv.size() > 0){
                res.push_back(tmv);
                tmv.clear();
            }
            while(!stack2.empty()){
                tmp = stack2.top();
                tmv.push_back(tmp->val);
                if(tmp->right)
                    stack1.push(tmp->right);
                if(tmp->left)
                    stack1.push(tmp->left);
                stack2.pop();
            }
            if(tmv.size() > 0){
                res.push_back(tmv);
                tmv.clear();
            }
        }
         return res;
    }
};