《剑指offer》面试题32.3：之字形打印二叉树

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

---

思路：

1.使用两个栈来解决这个问题。我们在打印某一层的节点时，把下一层的子节点保存到相应的栈里。
2.如果当前打印的是奇数层（第一层、第三层等），则先保存左子节点再保存右子节点到第一个栈里；如果当前打印的是偶数层（第二层、第四层等），则先保存右子节点再保存左子节点到第二个栈里。

java参考代码如下：

package chapter4;

import java.util.ArrayDeque;
import java.util.ArrayList;

public class P176_printTreeInSpecial {
    public static class TreeNode{
        int val;
        TreeNode left;
        TreeNode right;
        public TreeNode(int val){
            this.val=val;
            this.left=null;
            this.right=null;
        }
    }
    public static ArrayList<ArrayList<Integer>> printTreeInSpeical(TreeNode root){
        if(root==null) return null;
        ArrayList<ArrayList<Integer>> res=new ArrayList<>();
        ArrayDeque<TreeNode> stack1=new ArrayDeque<>();
        ArrayDeque<TreeNode> stack2=new ArrayDeque<>();
        stack1.addFirst(root);
        while (!stack1.isEmpty()||!stack2.isEmpty()){
            ArrayList<Integer> level=new ArrayList<>();
            for(int i=stack1.size();i>0;i--){
                TreeNode p=stack1.pollFirst();
                level.add(p.val);
                if(p.left!=null){//left-->right
                    stack2.addFirst(p.left);
                }
                if(p.right!=null){
                    stack2.addFirst(p.right);
                }
            }
            res.add(level);
            level=new ArrayList<>();
            for(int i=stack2.size();i>0;i--){
                TreeNode p=stack2.pollFirst();
                level.add(p.val);
                if(p.right!=null){//right-->left
                    stack1.addFirst(p.right);
                }
                if(p.left!=null){
                    stack1.addFirst(p.left);
                }
            }
            res.add(level);
        }
        return res;
    }

    public static void main(String[] args) {
        TreeNode root = new TreeNode(1);
        root.left = new TreeNode(2);
        root.right = new TreeNode(3);
        root.left.left = new TreeNode(4);
        root.left.right = new TreeNode(5);
        root.right.left = new TreeNode(6);
        root.right.right = new TreeNode(7);
        System.out.println(printTreeInSpeical(root));//之字形打印
    }
}

基于以上思路，参考代码如下：

class Solution {
public:
    void Print(TreeNode* pRoot) {
        if(pRoot == nullptr)
            return;
          
        stack<TreeNode*> levels[2];
        int current=0;
        int next=1;
        
        levels[current].push(pRoot);
        while(!levels[0].empty() || !levels[1].empty())
        {
        	TreeNode* pNode=levels[current].top();
        	levels[current].pop();

		printf("%d",pNode->value);

		if(current==0) //奇数行，先左后右
		{
			if(pNode->left!=nullptr)
				 levels[next].push(pNode->left);
			if(pNode->right!=nullptr)
				 levels[next].push(pNode->right);
		}
		else //偶数行，先右后左
		{
			if(pNode->right!=nullptr)
				 levels[next].push(pNode->right);	
			if(pNode->left!=nullptr)
				 levels[next].push(pNode->left);		
		}

		if(levels[current].empty())
		{
			printf("\n");
			current=1-current;
			next=1-next;
		}
        }
   }      
 }

测试用例：

a.功能测试（完全二叉树；所有节点只有左子树的二叉树；所有节点只有右子树的二叉树）。
b.特殊输入测试（二叉树根节点为nullptr指针；只有一个节点的二叉树）。