Best way to remove one arraylist elements from another arraylist

Игорь Рыбаков picture Игорь Рыбаков · May 23, 2016 · Viewed 8.1k times · Source

What is the best performance method in Java (7,8) to eliminate integer elements of one Arraylist from another. All the elements are unique in the first and second lists.

At the moment I know the API method removeall and use it this way:

tempList.removeAll(tempList2);

The problem appears when I operate with arraylists have more than 10000 elements. For example when I remove 65000 elements, the delay appears to be about 2 seconds. But I need to opperate with even more large lists with more than 1000000 elements.

What is the strategy for this issue?

Maybe something with new Stream API should solve it?

Answer

Marco13 picture Marco13 · May 23, 2016

tl;dr:

Keep it simple. Use

list.removeAll(new HashSet<T>(listOfElementsToRemove));

instead.


As Eran already mentioned in his answer: The low performance stems from the fact that the pseudocode of a generic removeAll implementation is

public boolean removeAll(Collection<?> c) {
    for (each element e of this) {
        if (c.contains(e)) {
            this.remove(e);
        }
    }
}

So the contains call that is done on the list of elements to remove will cause the O(n*k) performance (where n is the number of elements to remove, and k is the number of elements in the list that the method is called on).

Naively, one could imagine that the this.remove(e) call on a List might also have O(k), and this implementation would also have quadratic complexity. But this is not the case: You mentioned that the lists are specifically ArrayList instances. And the ArrayList#removeAll method is implemented to delegate to a method called batchRemove that directly operates on the underlying array, and does not remove the elements individually.

So all you have to do is to make sure that the lookup in the collection that contains the elements to remove is fast - preferably O(1). This can be achieved by putting these elements into a Set. In the end, it can just be written as

list.removeAll(new HashSet<T>(listOfElementsToRemove));

Side notes:

The answer by Eran has IMHO two major drawbacks: First of all, it requires sorting the lists, which is O(n*logn) - and it's simply not necessary. But more importantly (and obviously) : Sorting will likely change the order of the elements! What if this is simply not desired?

Remotely related: There are some other subtleties involved in the removeAll implementations. For example, HashSet removeAll method is surprisingly slow in some cases. Although this also boils down to the O(n*n) when the elements to be removed are stored in a list, the exact behavior may indeed be surprising in this particular case.