Binary Tree - Check if each node contains the sum of left & right subtree (Sum Tree)

Source Code (Explanation in above video)

package binarytree;

class Node {
  Node left;
  Node right;
  Node next;
  int data;
}

class BinaryTree {

  public boolean ifSumTree(Node node) {
    int leftSum;
    int rightSum;

    if (node == null || ifLeafNode(node))
      return true;

    if (ifSumTree(node.left&& ifSumTree(node.right)) {
      if (node.left == null)
        leftSum = 0;
      else if (ifLeafNode(node.left))
        leftSum = node.left.data;
      else
        leftSum = (node.left.data);

      if (node.right == null)
        rightSum = 0;
      else if (ifLeafNode(node.right))
        rightSum = node.right.data;
      else
        rightSum = (node.right.data);

      if ((node.data == rightSum + leftSum))
        return true;
      else
        return false;
    }

    return false;
  }

  public boolean ifLeafNode(Node node) {
    if (node == null) {
      return false;
    }

    if (node.left == null && node.right == null) {
      return true;
    }

    return false;
  }

  public Node createNewNode(int val) {
    Node newNode = new Node();
    newNode.data = val;
    newNode.left = null;
    newNode.next = null;
    newNode.right = null;
    return newNode;
  }
}

public class BinaryTreeApp {

  public static void main(String[] args) {

    BinaryTree a = new BinaryTree();

    Node root = a.createNewNode(120);
    root.left = a.createNewNode(40);
    root.left.left = a.createNewNode(15);
    root.left.right = a.createNewNode(25);
    root.right = a.createNewNode(20);
    root.right.right = a.createNewNode(20);

    System.out.println(a.ifSumTree(root));
  }
}

Leave a Reply

Your email address will not be published. Required fields are marked *