剑指offer计划6( 搜索与回溯算法简单版)---java
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;
}
}