Java encuentra la ruta donde la suma de los valores del nodo desde el nodo raíz hasta el nodo hoja es igual a la suma

Descripción del Título

Dado un árbol binario y una suma de valores, busque la ruta donde la suma de los valores del nodo desde el nodo raíz hasta el nodo hoja es igual a la suma,
por ejemplo:
Dado el siguiente árbol binario, suma = 22,

Devuelve
[
[5,4,11,2],
[5,8,9]
]

Ejemplo 1

entrar

{1,2}, 1

valor de retorno

[]

Ejemplo 2

entrar

{1,2}, 3

valor de retorno

[[1,2]]

 

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
    ArrayList<Integer> path = new ArrayList();
    
    /**
     * 
     * @param root TreeNode类 
     * @param sum int整型 
     * @return int整型ArrayList<ArrayList<>>
     */
    public ArrayList<ArrayList<Integer>> pathSum (TreeNode root, int sum) {
        preTree(root,sum);
        return result;
    }
    
    public void preTree(TreeNode root,int sum){
        
        if(root == null){
            return;
        }
        
        if(root.left == null && root.right == null && sum - root.val == 0){
            path.add(root.val);
            result.add(new ArrayList<Integer>(path));
            path.remove(path.size()-1);
            return;
        }
        
        path.add(root.val);
        preTree(root.left,sum - root.val);
        preTree(root.right,sum - root.val);
        path.remove(path.size()-1);
    }
}

 

Supongo que te gusta

Origin blog.csdn.net/luzhensmart/article/details/112968165
Recomendado
Clasificación