LeetCode – Ladrón de casas III (Java)

public int rob(TreeNode root) {
    if(root == null)
        return 0;
 
    int[] result = helper(root);
    return Math.max(result[0], result[1]);
}
 
public int[] helper(TreeNode root){
    if(root == null){
        int[] result = {0, 0};
        return result;
    }
 
    int[] result = new int[2];
    int[] left = helper(root.left);
    int[] right = helper (root.right);
 
    // result[0] is when root is selected, result[1] is when not. 
    result[0] = root.val + left[1] + right[1];
    result[1] = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);
 
    return result;
}

public int rob (TreeNode root) {if (root == null) return 0; En t[] resultado = ayudante (raíz); return Math.max (resultado[0], resultado[1]); } public int[] helper (TreeNode root) {if (root == null) {int[] resultado = {0, 0}; devolver resultado; } En t[] resultado = nuevo int[2]; En t[] left = helper (root.left); En t[] derecha = ayudante (root.right); // resultado[0] es cuando se selecciona la raíz, el resultado[1] es cuando no. resultado[0] = root.val + left[1] + derecha[1]; resultado[1] = Math.max (izquierda[0], izquierda[1]) + Math.max (derecha[0], derecho[1]); devolver resultado; }