剑指offer (59)

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

思路 : 1.使用两个栈S1和S2存储奇数层和偶数层节点
2.奇数层从左到右push进左右子树(出队时就可以是先右后左了)
3.偶数层从右到左push进左右子树(出队时就可以是先左后右)

	public static ArrayList<ArrayList<Integer>> print(TreeNode pRoot) {
		ArrayList<ArrayList<Integer>> res = new ArrayList<>();
		if (pRoot == null) {
			return res;
		}
		Stack<TreeNode> stack1 = new Stack<>();
		Stack<TreeNode> stack2 = new Stack<>();
		int deepth = 1;
		stack1.push(pRoot);
		while (!stack1.isEmpty() || !stack2.isEmpty()) {
			// 奇数层
			if (deepth % 2 != 0) {
				ArrayList<Integer> list1 = new ArrayList<>();//用来暂存结果
				while (!stack1.isEmpty()) {
					TreeNode cur = stack1.pop();
					list1.add(cur.value);
					if (cur.left != null) {
						stack2.push(cur.left);
					}
					if (cur.right != null) {
						stack2.push(cur.right);
					}
				}
				if (!list1.isEmpty()) {
					res.add(list1);
					deepth++;
				}
			}
			// 偶数层
			else {
				ArrayList<Integer> list2 = new ArrayList<>();//用来暂存结果
				while (!stack2.isEmpty()) {
					TreeNode cur = stack2.pop();
					list2.add(cur.value);
					if (cur.right != null) {
						stack1.push(cur.right);
					}
					if (cur.left != null) {
						stack1.push(cur.left);
					}
				}
				if (!list2.isEmpty()) {
					res.add(list2);
					deepth++;
				}
			}
		}
		return res;
	}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值