二叉树算法题目

题目一

折纸问题

请把一段纸条竖着放在桌子上，然后从纸条的下边向上方对折，压出折痕后展开。此时折痕是凹下去的，即折痕凸起的方向指向纸条的背面。如果从纸条的下边向上连续对折2次，压出折痕后展开，从上到下依次是下折痕、下折痕和上折痕。给定一个输入参数N，代表纸条都是从下边向上方连续对折N次，请从上到下打印所有折痕的方向。

例如：N=1时，打印：

down

N=2时，打印：

down

down

up

思路

二叉树的中序遍历

打印结果：down down up down down up up…

代码实现

public class PaperFolding{
    public static void printAllFolds(int N){
        printProcess(1,N,true);
    }
    
    public static void printProcess(int i,int N, boolean down){
        if(i>N){
            return;
        }
        printProcess(i + 1, N, true);
        System.out.println(down ? "down" : "up");
        printProcess(i + 1, N, false);
    }
}

题目二

从上往下打印出二叉树的每个节点，同层节点从左至右打印。

思路

二叉树的层次遍历问题，可以借助一个队列

代码实现

public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer> list = new ArrayList<Integer>();
        if(root==null){
            return list;
        }
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            TreeNode treeNode = queue.poll();
            if (treeNode.left != null) {
                queue.offer(treeNode.left);
            }
            if (treeNode.right != null) {
                queue.offer(treeNode.right);
            }
            list.add(treeNode.val);
        }
        return list;
    }
}

题目三

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。

链接

思路

前序遍历中，第一个是根节点的值，然后对比中序遍历，这个根节点的左边是左子树，右边是右子树，借助Arrays.copyOfRange()函数，用递归可以实现

代码实现

* public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
import java.util.Arrays;
public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        if(pre.length == 0 || in.length == 0){
            return null;
        }
        TreeNode root = new TreeNode(pre[0]);
        //先找到根节点
        for(int i = 0; i< in.length; i++){
            if(in[i] == pre[0]){
                //左子树 注意 copyOfRange 函数，左闭右开
                root.left = reConstructBinaryTree(Arrays.copyOfRange(pre,1,i+1),Arrays.copyOfRange(in,0,i));
                //右子树
                root.right = reConstructBinaryTree(Arrays.copyOfRange(pre,i+1,pre.length),Arrays.copyOfRange(in,i+1,in.length));
                break;
            }
        }
        return root;
    }
}