leetcode解题之103. Binary Tree Zigzag Level Order Traversal java版（之字形打印二叉树）

最新推荐文章于 2024-01-19 11:15:23 发布

mine_song

最新推荐文章于 2024-01-19 11:15:23 发布

阅读量509

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： java leetcode 103. Binary Tree Zig

本文链接：https://blog.youkuaiyun.com/mine_song/article/details/69940693

leetcode 专栏收录该内容

155 篇文章

订阅专栏

本文介绍了一种实现二叉树之字形打印的方法，通过使用两个栈来交替存储不同层级的节点，并根据层级的变化调整节点的添加顺序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

103. Binary Tree Zigzag Level Order Traversal

Given a binary tree, return the zigzag level order traversal of its nodes' values. (ie, from left to right, then right to left for the next level and alternate between).

For example:
Given binary tree [3,9,20,null,null,15,7],

return its zigzag level order traversal as:

[
  [3],
  [20,9],
  [15,7]
]

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

public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
		List<List<Integer>> ans = new ArrayList<>();
		if (root == null)
			return ans;
		// 使用两个栈维护顺序
		Stack<TreeNode> stack = new Stack<>();
		Stack<TreeNode> nextStack = new Stack<>();
		stack.add(root);
		int flag = 0;
		List<Integer> lay = new ArrayList<>();
		while (!stack.isEmpty()) {
			TreeNode node = stack.pop();
			lay.add(node.val);
			// 如果当前是从左到右遍历，按左子树右子树的顺序添加
			if (flag == 0) {
				if (node.left != null)
					nextStack.add(node.left);
				if (node.right != null)
					nextStack.add(node.right);
			} else// 如果当前是从右到左遍历，按右子树左子树的顺序添加
			{
				if (node.right != null)
					nextStack.add(node.right);
				if (node.left != null)
					nextStack.add(node.left);
			}
			if (stack.isEmpty()) {
				// 交换两个栈
				Stack<TreeNode> tmp = stack;
				stack = nextStack;
				nextStack = tmp;
				// 标记下一层处理的方向
				flag = 1 - flag;
				ans.add(new ArrayList<>(lay));
				lay.clear();
			}
		}
		return ans;

	}