Tree Sum is a type of Binary Tree where every node contains the sum of values of nodes in the left and right subtrees in the original Binary Tree.
package org.wesome.dsalgo;
import lombok.Data;
import java.util.Objects;
public class BinarySearchTree {
Node root;
int treeSum(Node node) {
if (Objects.isNull(node)) {
return 0;
}
int oldVal = node.data;
node.data = treeSum(node.left) + treeSum(node.right);
return node.data + oldVal;
}
/* Inorder -> Left -> Root -> Right */
void printInOrder() {
System.out.println("print In Order");
printInOrder(root);
}
void printInOrder(Node root) {
if (Objects.isNull(root)) {
return;
}
printInOrder(root.left);
System.out.print(" " + root.data);
printInOrder(root.right);
}
void insert(int data) {
System.out.println("BinarySearchTree.insert data = " + data);
root = insert(root, data);
}
Node insert(Node root, int data) {
if (Objects.isNull(root)) {
Node tempNode = new Node(data);
return tempNode;
}
if (data > root.data) {
root.right = insert(root.right, data);
} else {
root.left = insert(root.left, data);
}
return root;
}
}
@Data
class Node {
int data;
Node left, right;
public Node(int data) {
this.data = data;
left = right = null;
}
}
package org.wesome.dsalgo;
import org.junit.jupiter.api.Test;
public class BinarySearchTreeTest {
@Test
void treeSumTest() {
BinarySearchTree binarySearchTree = new BinarySearchTree();
binarySearchTree.insert(4);
binarySearchTree.insert(2);
binarySearchTree.insert(1);
binarySearchTree.insert(3);
binarySearchTree.insert(6);
binarySearchTree.insert(5);
binarySearchTree.insert(7);
binarySearchTree.treeSum(binarySearchTree.root);
binarySearchTree.printInOrder();
}
}
plugins {
id 'java'
id "io.freefair.lombok" version "6.2.0"
}
group = 'org.wesome'
version = '0.0.1-SNAPSHOT'
sourceCompatibility = JavaVersion.VERSION_1_8
configurations {
compileOnly {
extendsFrom annotationProcessor
}
}
repositories {
mavenCentral()
}
dependencies {
testImplementation 'org.junit.jupiter:junit-jupiter:5.6.2'
}
test {
useJUnitPlatform()
}