I am currently working on a Foobar challenge, the Ion Flux Relabelling. I came up with a java solution but Google returns an OutOfMemoryError
.
The prompt is, quoting the original question:
Oh no! Commander Lambda's latest experiment to improve the efficiency of her LAMBCHOP doomsday device has backfired spectacularly. She had been improving the structure of the ion flux converter tree, but something went terribly wrong and the flux chains exploded. Some of the ion flux converters survived the explosion intact, but others had their position labels blasted off. She's having her henchmen rebuild the ion flux converter tree by hand, but you think you can do it much more quickly - quickly enough, perhaps, to earn a promotion!
Flux chains require perfect binary trees, so Lambda's design arranged the ion flux converters to form one. To label them, she performed a post-order traversal of the tree of converters and labeled each converter with the order of that converter in the traversal, starting at 1. For example, a tree of 7 converters would look like the following:
7 / \ 3 6 / \ / \ 1 2 4 5
Write a function answer(h, q) - where h is the height of the perfect tree of converters and q is a list of positive integers representing different flux converters - which returns a list of integers p where each element in p is the label of the converter that sits on top of the respective converter in q, or -1 if there is no such converter. For example, answer(3, [1, 4, 7]) would return the converters above the converters at indexes 1, 4, and 7 in a perfect binary tree of height 3, which is [3, 6, -1].
The domain of the integer h is 1 <= h <= 30, where h = 1 represents a perfect binary tree containing only the root, h = 2 represents a perfect binary tree with the root and two leaf nodes, h = 3 represents a perfect binary tree with the root, two internal nodes and four leaf nodes (like the example above), and so forth. The lists q and p contain at least one but no more than 10000 distinct integers, all of which will be between 1 and 2^h-1, inclusive.
The test cases provided are:
Inputs:
(int) h = 3
(int list) q = [7, 3, 5, 1]
Output:
(int list) [-1, 7, 6, 3]
Inputs:
(int) h = 5
(int list) q = [19, 14, 28]
Output:
(int list) [21, 15, 29]
I used a rather complex and dumb way to solve the problem. I created two lists, one recording every pair of children in the binary tree, the other one recording all the corresponding parents to these children in corresponding indexes. So my children list for the height 3 binary tree provided above would look like [[3, 6], [1, 2], [4, 5]]
, and the relevant part of my parent list would look like [7, 3, 6]
.
To find the parent of the elements given in array q
, I just ran a loop to look for the element in the children list and, after finding it, record the corresponding parent since they have the same index.
import java.util.*;
public class Answer {
public static int[] answer(int h, int[] q) {
int root = ((int) Math.pow(2, h) - 1);
int[] corr = new int[q.length];
List<List<Integer>> childs = new ArrayList<List<Integer>>();
List<Integer> parent = new ArrayList<Integer>();
parent.add(root);
//Loop through each "height"
for(int i = 0; i < h - 1; i++){
int index = 0;
index = parent.size();
//Loop through each element in each height
for(int l = (int) Math.pow(2, i); l > 0; l--){
int x = parent.get(index - l) - (int) Math.pow(2, h-(i+1));
int y = parent.get(index - l) - 1;
childs.add(Arrays.asList(x, y));
parent.add(x);
parent.add(y);
}
}
for(int i = 0; i < q.length; i++){
if(q[i] == root){
corr[i] = -1;
}
for(int l = 0; l < childs.size(); l++){
if(childs.get(l).get(0) == q[i] || childs.get(l).get(1) == q[i]){
corr[i] = parent.get(l);
}
}
}
return corr;
}
}
Like I said, Google returns an OutOfMemoryError
for my code during execution.
I totally understand that my method is not the most efficient solution. So really, I am trying to find the most effective way to solve this problem in Java. There must be something really simple that I am just not seeing.