Amazon Interview Question
Software DevelopersCountry: United States
Interview Type: In-Person
public class SmallerOnRight {
public static void updateCountSmaller(int[] elements, int[] smallerCount) {
BSTNode root = new BSTNode(elements[0], 0);
for (int i = 1; i < elements.length; i++) {
BSTNode parent = null;
BSTNode temp = root;
while (temp != null) {
if (temp.getData() > elements[i]) {
parent = temp;
updateSmallerCount(parent, smallerCount);
temp = temp.getLeft();
if (temp == null) {
parent.setLeft(new BSTNode(elements[i], i));
}
} else {
parent = temp;
temp = temp.getRight();
if (temp == null) {
parent.setRight(new BSTNode(elements[i], i));
}
}
}
}
}
public static void updateSmallerCount(BSTNode parent, int[] smallerCount) {
BSTNode runner = parent;
while (runner != null) {
smallerCount[runner.getIndex()] = smallerCount[runner.getIndex()] + 1;
runner = runner.getRight();
}
}
public static void main(String[] args) {
int[] elements = new int[] { 12, 1, 2, 3, 0, 11, 4 };
int[] smallerCount = new int[] { 0, 0, 0, 0, 0, 0, 0 };
updateCountSmaller(elements, smallerCount);
}
}
use a balanced binary search tree, similar to geeksforgeeks.org/count-smaller-elements-on-right-side
- confused_coder July 25, 2016