按之字形顺序打印二叉树

题目描述

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

思路：用两个栈交替压入弹出处理，代码如下：

 vector<vector<int> > Print(TreeNode* pRoot) {
        vector<vector<int>> vv;
        if(pRoot == NULL)
            return vv;
        stack<TreeNode*> s[2];
        int current = 0;
        int next = 1;
        s[0].push(pRoot);
        vector<int> v;
        while(!s[0].empty() || !s[1].empty()){
            TreeNode* pNode = s[current].top();
            v.push_back(s[current].top()->val);
            s[current].pop();
            if(current == 0){
                if(pNode->left)
                    s[next].push(pNode->left);
                if(pNode->right)
                    s[next].push(pNode->right);
                
            }else{
                if(pNode->right)
                    s[next].push(pNode->right);
                if(pNode->left)
                    s[next].push(pNode->left);
            }
            if(s[current].empty()){
                vv.push_back(v);
                v.clear();
                current = 1 - current;
                next = 1 - next;
            }
            
        }
        return vv;
    }