剑指offer:把二叉树按照之字形打印

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

public static ArrayList<ArrayList<Integer> > PrintByZhi(TreeNode pRoot) {
		ArrayList<ArrayList<Integer>> arrs = new ArrayList<ArrayList<Integer> >();
		if(pRoot == null)
		{
			return arrs;
		}
		//定义两个栈
		Stack<TreeNode> stack1 = new Stack<TreeNode>();
		Stack<TreeNode> stack2 = new Stack<TreeNode>();
		stack1.push(pRoot);
		TreeNode node = null;
		while(!stack1.isEmpty() || !stack2.isEmpty())
		{
			ArrayList<Integer>list = new ArrayList<Integer>();
			if(!stack1.isEmpty())
			{
				while(!stack1.isEmpty())
				{
					node = stack1.pop();
					list.add(node.value);
					if(node.leftNode != null)
					{
						stack2.push(node.leftNode);
					}
					if(node.rightNode != null)
					{
						stack2.push(node.rightNode);
					}
				}
				arrs.add(list);
			}
			else
			{
				while(!stack2.isEmpty())
				{
					node = stack2.pop();
					list.add(node.value);
					if(node.rightNode != null)
					{
						stack1.push(node.rightNode);
					}
					if(node.leftNode!=null)
					{
						stack1.push(node.leftNode);
					}
				}
				arrs.add(list);
			}
		}
		return arrs;
	}

猜你喜欢

转载自blog.csdn.net/rodman177/article/details/89741795