Oferta reconstrucción de un árbol binario para probar la seguridad

título Descripción

Y escriba el resultado en un orden previo de recorrido de un árbol binario en preorden recorrido del árbol binario una reconstrucción. Resultados Supongamos que preorden de recorrido de orden y de entrada de números duplicados son libres. Por ejemplo, antes de entrar en preorden recorrido secuencia {1,2,4,7,3,5,6,8} y {4,7,2,1,5,3,8,6} secuencia de orden transversal, y el árbol binario reconstruida regresar.

pensamiento

Sabemos que el primer nodo en un recorrido en preorden es la raíz del árbol, por lo que vamos a crear la raíz de acuerdo con el preámbulo de la primera secuencia de dígitos de recorrido, a continuación, encontrar la ubicación del nodo raíz en la secuencia de preventa, raíz izquierda sub-árbol está a la izquierda, derecha es el sub-árbol derecho, por lo que podemos determinar el número de nodo del sub-árbol de la izquierda y la derecha. Después de dividir los valores de la izquierda, derecha subárbol nodo y secuencia preorder traversal orden de la parte frontal, puede ser construido de forma recursiva a cada uno de su izquierda y subárboles correctas.

Código de referencia

public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        
        // 终止条件
        if (pre == null || pre.length == 0) {
            return null;
        }
        
        // 创建根节点
        TreeNode root = new TreeNode(pre[0]);
        
        // 获取到根节点在中序数组的索引
        int index = findIndex(pre, in);
        
        // 前序数组左子树的索引为[1, index + 1), 中序数组左子树的索引为[0, index)
        root.left = reConstructBinaryTree(Arrays.copyOfRange(pre, 1, index + 1), Arrays.copyOfRange(in, 0, index));

        // 前序数组右子树的索引为[index + 1, pre.length), 中序数组右子树的索引为[index + 1, in.length)
        root.right = reConstructBinaryTree(Arrays.copyOfRange(pre, index + 1, pre.length), Arrays.copyOfRange(in, index + 1, pre.length));
        
        return root;
    }
    
    private int findIndex(int [] pre, int [] in) {
        for (int i = 0; i < in.length; i++) {
            if (in[i] == pre[0]) {
                return i;
            }
        }
        return -1;
    }
}