package org.wesome.dsalgo;
import lombok.Data;
import java.util.LinkedList;
import java.util.Map;
import java.util.Objects;
import java.util.Queue;
import java.util.TreeMap;
@Data
class Node {
int data;
int hrzDis;
Node left, right;
public Node(int data) {
this.data = data;
left = right = null;
}
}
public class BinarySearchTree {
Node root;
public static void bottomView(Node root) {
if (Objects.isNull(root)) {
return;
}
int hrzDis = 0;
Map<Integer, Integer> map = new TreeMap<>();
// First add nodes in Queue for level order traversal
Queue<Node> queue = new LinkedList<>();
// Assign initialized horizontal distance value to root node and add it to the queue.
root.hrzDis = hrzDis;
queue.add(root);
while (!queue.isEmpty()) {
Node node = queue.remove();
hrzDis = node.hrzDis;
map.put(hrzDis, node.data);
// add left child in the queue with horizontal distance hrzDis-1.
if (Objects.nonNull(node.left)) {
node.left.hrzDis = hrzDis - 1;
queue.add(node.left);
}
// add right child in the queue with horizontal distance hrzDis+1.
if (Objects.nonNull(node.right)) {
node.right.hrzDis = hrzDis + 1;
queue.add(node.right);
}
}
for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
System.out.print(entry.getValue() + " ");
}
}
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;
}
}
package org.wesome.dsalgo;
import org.junit.jupiter.api.Test;
public class BinarySearchTreeTest {
@Test
void bottomViewTest() {
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.insert(7);
binarySearchTree.bottomView(binarySearchTree.root);
}
}
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()
}