按之字形顺序打印二叉树 java

最新推荐文章于 2023-05-08 10:01:36 发布

原创最新推荐文章于 2023-05-08 10:01:36 发布 · 640 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

38 篇文章

订阅专栏

本文介绍了一种实现之字形打印二叉树的方法，通过使用两个栈交替存放节点来实现从左到右和从右到左的打印顺序转换。此算法适用于需要按特定顺序遍历二叉树的应用场景。

题目描述

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

public class Solution {
		public ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
			ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
			if (pRoot == null) {
				return list;
			}
			//创建两个栈进行分别存储
			Stack<TreeNode>[] stacks = new Stack[2];
			stacks[0] = new Stack<TreeNode>();
			stacks[1] = new Stack<TreeNode>();
			int current = 0;
			int next = 1;
			stacks[0].add(pRoot);
			ArrayList<Integer> array=new ArrayList<Integer>();
			while (!stacks[0].isEmpty() || !stacks[1].isEmpty()) {
				TreeNode node=stacks[current].pop();
				array.add(node.val);
				//从左往右压入下一个栈中，在读取下一个栈的时候是从右往左的
				if(current==0){
					if(node.left!=null){
						stacks[next].add(node.left);
					}
					if(node.right!=null){
						stacks[next].add(node.right);
					}
				//从右往左压入下一个栈中，在读取下一个栈的时候是从左往有的
				}else{
					if(node.right!=null){
						stacks[next].add(node.right);
					}
					if(node.left!=null){
						stacks[next].add(node.left);
					}
				}
				//遍历完一个栈之后进行遍历另一个栈
				if(stacks[current].isEmpty()){
					current=1-current;
					next=1-next;
					list.add(array);
					array=new ArrayList<Integer>();
				}
				
			}
			return list;
		}

	}