【剑指offer】按层和按之字打印二叉树
来源:互联网 发布:数字规律软件 编辑:程序博客网 时间:2024/05/21 09:56
这是网上思路比较清晰的广度优先遍历解法。
ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) { ArrayList<ArrayList<Integer>> result=new ArrayList<ArrayList<Integer>>(); if(pRoot==null) return result; Queue<TreeNode> queue=new LinkedList<TreeNode>(); ArrayList<Integer> layer=new ArrayList<Integer>(); queue.add(pRoot); int start=0,end=queue.size(); while(!queue.isEmpty()){ TreeNode tmp=queue.remove(); layer.add(tmp.val); start++; if(tmp.left!=null) queue.add(tmp.left); if(tmp.right!=null) queue.add(tmp.right); if(start==end){ end=queue.size(); start=0; result.add(layer); layer=new ArrayList<Integer>(); } } return result; }
按之字形打印二叉树时,书上说的是用两个栈。但是首先想到的是偶数层用队列,奇数层用栈(根结点属于0层)。然而结果证明这种方法不行,因为在打印奇数层时它是从右至左出栈的,这时将结点的左右子结点放入下一层偶数层的队列时,顺序是右左,右右,左左,左右。即使改成先右子结点再左子结点也不行,因为右右,右左,左右,左左仍然是反的。同样,用两个队列也无法实现按之字形打印。
根据BFS可以很快写出代码:
public ArrayList<ArrayList<Integer> > print(TreeNode pRoot) { ArrayList<ArrayList<Integer>> result=new ArrayList<ArrayList<Integer>>(); if(pRoot==null) return result; Stack<TreeNode> leftToRight=new Stack<TreeNode>(); Stack<TreeNode> rightToLeft=new Stack<TreeNode>(); ArrayList<Integer> layer=new ArrayList<Integer>(); leftToRight.add(pRoot); int start=0,end=leftToRight.size(); TreeNode tmp; while(!leftToRight.isEmpty()||!rightToLeft.isEmpty()){ while(!leftToRight.isEmpty()){ tmp=leftToRight.pop(); layer.add(tmp.val); start++; if(tmp.left!=null)rightToLeft.push(tmp.left); if(tmp.right!=null)rightToLeft.push(tmp.right); if(start==end){ start=0; end=rightToLeft.size(); result.add(layer); layer=new ArrayList<Integer>(); } } while(!rightToLeft.isEmpty()){ tmp=rightToLeft.pop(); layer.add(tmp.val); start++; if(tmp.right!=null)leftToRight.add(tmp.right); if(tmp.left!=null)leftToRight.add(tmp.left); if(start==end){ start=0; end=leftToRight.size(); result.add(layer); layer=new ArrayList<Integer>(); } } } return result; }
上面的代码可以更简化
0 0
- 【剑指offer】按层和按之字打印二叉树
- [剑指offer]按层打印二叉树
- 剑指offer:按层打印二叉树
- 《剑指offer》按之字行顺序打印二叉树
- 剑指offer(五十三)之按之字形顺序打印二叉树
- java面试题(剑指offer):按层打印二叉树
- 二叉树的按层打印和ZigZag打印
- 按层打印二叉树
- 按层打印二叉树
- 按层打印二叉树
- 按层打印二叉树
- 按层打印二叉树
- 按层打印二叉树
- 剑指 offer:从上到下打印二叉树(层序遍历)
- 剑指offer 打印二叉树路径和
- 剑指offer--安之字顺序打印二叉树
- 【剑指offer】题61:之字打印二叉树
- 【刷题之路】按层打印二叉树
- Java调用MySQL的存储过程
- PHP的小技巧
- 《连载 | 物联网框架ServerSuperIO教程》- 14.配制工具介绍,以及设备驱动、视图驱动、服务实例的挂载
- Tips: Create An Empty Table Has Same Columns with Existed Table
- 【OI做题记录】【BZOJ】【Usaco2008 Mar】土地购买
- 【剑指offer】按层和按之字打印二叉树
- 开源物联网框架ServerSuperIO 3.0正式发布(C#),跨平台:Win&Win10 Iot&Ubuntu&Ubuntu Mate,一套设备驱动跨平台挂载,附:开发套件和教程。
- 计算机导论1
- Interlude: Troubleshooting for 'Invalid object name'
- Android 属性动画
- EL函数库
- 两数之和(三)
- perl use base 继承
- Spring Boot 配置优先级顺序