Enough is enough!
Alice and Bob were on a holiday. Both of them took many pictures of the places they've been, and now they want to show Charlie their entire collection. However, Charlie doesn't like this sessions, since the motive usually repeats. He isn't fond of seeing the Eiffel tower 40 times. He tells them that he will only sit during the session if they show the same motive at most N times. Luckily, Alice and Bob are able to encode the motive as a number. Can you help them to remove numbers such that their list contains each number only up to N times, without changing the order?
Task
Given a list lst and a number N, create a new list that contains each number of lst at most N times without reordering. For example if N = 2, and the input is [1,2,3,1,2,1,2,3], you take [1,2,3,1,2], drop the next [1,2] since this would lead to 1 and 2 being in the result 3 times, and then take 3, which leads to [1,2,3,1,2,3].
Example
EnoughIsEnough.deleteNth(new int[] {20,37,20,21}, 1) // return [20,37,21]
EnoughIsEnough.deleteNth(new int[] {1,1,3,3,7,2,2,2,2}, 3) // return [1, 1, 3, 3, 7, 2, 2, 2]
Good Solution1:
import java.util.*;
public class EnoughIsEnough {
public static int[] deleteNth(int[] elements, int max) {
if (max < 1) return new int[0];
final HashMap<Integer,Integer> map = new HashMap<>();
final List<Integer> list = new ArrayList<>();
for (final Integer i : elements) {
final Integer v = map.put(i, map.getOrDefault(i, 0) + 1);
if (v == null || v < max) list.add(i);
}
return list.stream().mapToInt(i->i).toArray();
}
}
Good Solution2:
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
class EnoughIsEnough {
private static boolean shouldAdd(final Map<Integer, Integer> counts, final int element, final int maxOcurrences) {
if (counts.getOrDefault(element, 0) < maxOcurrences) {
counts.merge(element, 1, Integer::sum);
return true;
}
return false;
}
static int[] deleteNth(final int[] elements, final int maxOcurrences) {
final Map<Integer, Integer> counts = new HashMap<>();
return Arrays.stream(elements)
.filter(element -> shouldAdd(counts, element, maxOcurrences))
.toArray();
}
}