1.1、題目1
劍指 Offer 32 - I. 從上到下列印二叉樹
1.2、解法
其實這三道題都是廣度遍歷二叉樹的方式，
通過佇列實作，存進陣列中回傳，
1.3、代碼
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public int[] levelOrder(TreeNode root) {
if(root==null) return new int[0];
Queue<TreeNode> queue = new LinkedList<>(){{ add(root);}};
ArrayList<Integer> array = new ArrayList<Integer>();
while(!queue.isEmpty()){
TreeNode node = queue.poll();
array.add(node.val);
if(node.left!=null) queue.add(node.left);
if(node.right!=null) queue.add(node.right);
}
int[]res= new int[array.size()];
for(int i=0;i<array.size();i++){
res[i]=array.get(i);
}
return res;
}
}
2.1、題目2
劍指 Offer 32 - II. 從上到下列印二叉樹 II
2.2、解法
這題改了改，陣列需要分層次，就在while里定義一個List，到后面存進大的List中回傳即可，這題答案是直接copy的~~
2.3、代碼
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<>();
List<List<Integer>> res = new ArrayList<>();
if(root != null) queue.add(root);
while(!queue.isEmpty()) {
List<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);
}
res.add(tmp);
}
return res;
}
}
3.1、題目3
劍指 Offer 32 - III. 從上到下列印二叉樹 III
3.2、解法
這題是通過兩端添加，其實就是上題的稍微修改版本，這題答案是直接copy的~~
3.3、代碼
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<>();
List<List<Integer>> res = new ArrayList<>();
if(root != null) queue.add(root);
while(!queue.isEmpty()) {
LinkedList<Integer> tmp = new LinkedList<>();
for(int i = queue.size(); i > 0; i--) {
TreeNode node = queue.poll();
if(res.size() % 2 == 0) tmp.addLast(node.val); // 偶數層 -> 佇列頭部
else tmp.addFirst(node.val); // 奇數層 -> 佇列尾部
if(node.left != null) queue.add(node.left);
if(node.right != null) queue.add(node.right);
}
res.add(tmp);
}
return res;
}
}
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/297991.html
標籤：其他
上一篇：Java基礎之SPI機制
下一篇：AOP：面向切片編程