牛客题霸NC80把二叉树打印成多行Java题解
https://www.nowcoder.com/practice/445c44d982d04483b04a54f298796288?tpId=117&&tqId=35002&rp=1&ru=/ta/job-code-high&qru=/ta/job-code-high/question-ranking
方法:利用队列
解题思路:将节点加入到队列中,利用队列Queue的先进后出,依此弹出节点。如果弹出的节点有左、右子节点,则将左、右子节点加入到队列Queue中,将每一层的节点保存到一个ArrayList中(tmp),每一层遍历完,将这一层的节点都加入到ArrayList<ArrayList<integer>> list 中。</integer>
import java.util.ArrayList; import java.util.LinkedList; import java.util.Queue; /* public class TreeNode { int val = 0; TreeNode left = null; TreeNode right = null; public TreeNode(int val) { this.val = val; } } */ public class Solution { ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) { ArrayList<ArrayList<Integer>> list = new ArrayList<>(); Queue<TreeNode> queue = new LinkedList<>(); if(pRoot!=null){ queue.add(pRoot); } while(!queue.isEmpty()){ ArrayList<Integer> tmp = new ArrayList<>(); for(int i = queue.size();i>0;i--){ TreeNode node = queue.poll(); tmp.add(node.val); if(node.left!=null){ queue.add(node.left); } if(node.right!=null){ queue.add(node.right); } } list.add(tmp); } return list; } }