目录
一 二叉树的层序遍历
给你二叉树的根节点 root
,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)
思路:
这里我们创建一个list0 = new List<List<Integer>>,和一个list = new List<Integer>,再创建一个que = new Queue<Integer>;先将root放入que中,然后使用while循环当que为空时停下来,每释放一个que中的root,将其加入list;将这个root的左节点和右节点再进去que;然后每结束一个循环再将list加入list0。
代码:
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> sec= new ArrayList<>();
if(root == null){
return sec;
}
Queue<TreeNode> que = new LinkedList<>();
que.offer(root);
while(!que.isEmpty()){
int size = que.size();
List<Integer> fir = new ArrayList<>();
for (int i = 0; i < size; i++) {
TreeNode node = que.remove();
fir.add(node.val);
if(node.left != null){
que.offer(node.left);
}
if(node.right != null){
que.offer(node.right);
}
}
sec.add(fir);
}
return sec;
}
}
链接:
二 从前序和中序遍历中构造二叉树
给定两个整数数组 preorder 和 inorder ,其中 preorder 是二叉树的先序遍历, inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。
思路:写这种代码我们要知道:我们平时根据前序和中序遍历很容易推导出一个二叉树来,所以我们要做的就是将我们脑海中的过程转变成代码
首先前序遍历的第一个肯定是根节点,然后是先先往它的左节点走;但中序遍历的特点就是:在一个范围内,每个父节点的左边是它的左子树,而右边是它的右子树,所以我们要推断出这个范围;我们根据这两个特点就能够写出这个代码
比如:
根据preorder可知3是根节点,所以3的左边和右边分别是它的左子树和右子树
然后preorder的下一个是9,如果9在inorder的左边的话它就是3的左节点,相反就不是
代码:
class Solution {
//i用于记录preorder的位置
static int i;
public TreeNode buildTree(int[] preorder, int[] inorder) {
TreeNode root = new TreeNode();
root = buildTreeChild(preorder, inorder, 0, preorder.length-1);
//因为测试有多组测试,所以每次要将i置为0
i = 0;
return root;
}
private TreeNode buildTreeChild(int[] preorder, int[] inorder, int ib, int ie){
if(ib > ie || ib == -1){
return null;
}
TreeNode root = new TreeNode(preorder[i]);
//查找preorder[i]在inorder中的位置
int pos = findIndex(inorder, preorder[i], ib, ie);
i++;
root.left = buildTreeChild(preorder, inorder, ib, pos-1);
root.right = buildTreeChild(preorder, inorder, pos+1, ie);
return root;
}
private int findIndex(int[] inorder, int val , int ib, int ie){
int k= ib;
for(k = ib; k <= ie; k++){
if(val == inorder[k]){
return k;
}
}
//返回-2是因为在执行root.right时如果没有找到让ib=-1从而return null
return -2;
}
}
链接:
三 中序和后序遍历构造二叉树
思路:
根据上一题我们已经有做这种题目的经验了,inorder还是根据范围来判断左子树和右子树。但postorder和preorder不一样的是它是最后一个节点是root,然后在postorder中往前走,根据范围来判断root的右子树的左子树。
根据postorder 3是根节点,inorder中3的左右分别是根节点的左子树和右子树
然后postorder往前走,20在3的右边,所以20是3的右节点
代码:
class Solution {
static int i;
public TreeNode buildTree(int[] inorder, int[] postorder) {
TreeNode root = new TreeNode();
i = postorder.length-1;
root = buildTreeChild(inorder, postorder, 0, i);
i = postorder.length-1;
return root;
}
private TreeNode buildTreeChild(int[] inorder, int[] postorder, int ib, int ie){
if(ib > ie){
return null;
}
TreeNode root = new TreeNode(postorder[i]);
int pos = findIndex(inorder, postorder[i], ib, ie);
i--;
root.right = buildTreeChild(inorder, postorder, pos+1, ie);
root.left = buildTreeChild(inorder, postorder, ib, pos-1);
return root;
}
private int findIndex(int[] inorder, int val , int ib, int ie){
int i = ib;
for(i = ib; i <= ie; i++){
if(val == inorder[i]){
return i;
}
}
return 0;
}
}
链接:
注:题目均来自力扣