leetcode 102. 二叉树的层序遍历

方法一：
用两个数组进行迭代遍历

1. class Solution {
2.     public List<List<Integer>> levelOrder(TreeNode root) {
3.         if(root == null) {
4.             return new ArrayList<List<Integer>>();
5.         }
6.         List<TreeNode> cur = new ArrayList<TreeNode>();
7.         cur.add(root);
9.         List<List<Integer>> ans = new ArrayList<List<Integer>>();
10.         while (!cur.isEmpty()) {
11.             List<TreeNode> nxt = new ArrayList<TreeNode>();
12.             List<Integer> vals = new ArrayList<Integer>();
13.             for(TreeNode node : cur) {
14.                 vals.add(node.val);
15.                 if (node.left != null) {
16.                     nxt.add(node.left);
17.                 }
18.                 if (node.right != null) {
19.                     nxt.add(node.right);
20.                 }
21.             }
22.             cur = nxt;
23.             ans.add(vals);
24.         }
25.         return ans;
26.     }
27. }

复制代码

方法二：
只用到了一个队列

1. class Solution {
2.     public List<List<Integer>> levelOrder(TreeNode root) {
3.         if (root == null) {
4.             return new ArrayList<List<Integer>>();
5.         }
6.         Queue<TreeNode> q = new LinkedList<TreeNode>();
7.         List<List<Integer>> ans = new ArrayList<List<Integer>>();
8.         q.add(root);
9.         while (!q.isEmpty()) {
10.             int len = q.size();
11.             List<Integer> vals = new ArrayList<Integer>();
12.             while (len-- > 0) {
13.                 TreeNode node = q.poll();
14.                 if (node.left != null) {
15.                     q.add(node.left);
16.                 }
17.                 if (node.right != null) {
18.                     q.add(node.right);
19.                 }
20.                 vals.add(node.val);
21.             }
22.             ans.add(vals);
23.         }
24.         return ans;
25.     }
26. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。