Uzyskaj rozmiar iterowalnego w Javie

Muszę obliczyć liczbę elementów Iterablew języku Java. Wiem, że potrafię to zrobić:

Iterable values = ...
it = values.iterator();
while (it.hasNext()) {
  it.next();
  sum++;
}

Mógłbym też zrobić coś takiego, ponieważ nie potrzebuję już obiektów w Iterable:

it = values.iterator();
while (it.hasNext()) {
  it.remove();
  sum++;
}

Benchmark na małą skalę nie wykazał dużej różnicy w wydajności, żadnych komentarzy ani innych pomysłów na ten problem?

TL; DR: Użyj metody narzędziowej Iterables.size(Iterable)wielkiej biblioteki Guava .

Z dwóch fragmentów kodu powinieneś użyć pierwszego, ponieważ drugi usunie wszystkie elementy z values, więc później jest pusty. Zmiana struktury danych dla prostego zapytania, takiego jak jego rozmiar, jest bardzo nieoczekiwana.

Wydajność zależy od struktury danych. Jeśli jest to na przykład w rzeczywistości a ArrayList, usuwanie elementów od początku (co robi twoja druga metoda) jest bardzo powolne (obliczenie rozmiaru staje się O (n * n) zamiast O (n) tak, jak powinno).

Ogólnie rzecz biorąc, jeśli istnieje szansa, że valuesfaktycznie jest to a, Collectiona nie tylko Iterable, sprawdź to i zadzwoń size()w przypadku:

if (values instanceof Collection<?>) {
  return ((Collection<?>)values).size();
}
// use Iterator here...

Wezwanie do size()woli zwykle znacznie szybciej niż zliczanie liczby elementów, a ten trik jest dokładnie to, co Iterables.size(Iterable)z Guava robi dla Ciebie.

Jeśli pracujesz z Javą 8, możesz użyć:

Iterable values = ...
long size = values.spliterator().getExactSizeIfKnown();

zadziała tylko wtedy, gdy iterowalne źródło ma określony rozmiar. Większość Spliterators for Collections to zrobić, ale możesz mieć problemy, jeśli pochodzą one z HashSetlub ResultSetna przykład.

Możesz sprawdzić javadoc tutaj.

Jeśli Java 8 nie wchodzi w grę lub nie wiesz, skąd pochodzi iterowalność, możesz zastosować to samo podejście co guava:

  if (iterable instanceof Collection) {
        return ((Collection<?>) iterable).size();
    } else {
        int count = 0;
        Iterator iterator = iterable.iterator();
        while(iterator.hasNext()) {
            iterator.next();
            count++;
        }
        return count;
    }

To może trochę za późno, ale może komuś pomóc. Natknąłem się na podobny problem Iterablew mojej bazie kodu i rozwiązaniem było użycie for eachbez jawnego wywołania values.iterator();.

int size = 0;
for(T value : values) {
   size++;
}

Możesz przesłać swoją iterowalną listę na listę, a następnie użyć na niej .size ().

Lists.newArrayList(iterable).size();

Dla jasności powyższa metoda będzie wymagać następującego importu:

import com.google.common.collect.Lists;

Ściśle mówiąc, Iterable nie ma rozmiaru. Pomyśl o strukturze danych jak o cyklu.

I pomyśl o następującej iterowalnej instancji, bez rozmiaru:

    new Iterable(){

        @Override public Iterator iterator() {
            return new Iterator(){

                @Override
                public boolean hasNext() {
                    return isExternalSystemAvailble();
                }

                @Override
                public Object next() {
                    return fetchDataFromExternalSystem();
                }};
        }};

Wybrałbym it.next()z prostego powodu, który next()jest gwarantowany do zaimplementowania, podczas gdy remove()jest to operacja opcjonalna.

E next()

Zwraca następny element w iteracji.

void remove()

Usuwa z podstawowej kolekcji ostatni element zwrócony przez iterator (operacja opcjonalna) .

java 8 i nowsze

StreamSupport.stream(data.spliterator(), false).count();

Jak dla mnie to tylko różne metody. Pierwsza z nich pozostawia obiekt, na którym iterujesz, bez zmian, podczas gdy sekundy pozostawiają go pustym. Pytanie brzmi, co chcesz robić. Złożoność usuwania opiera się na implementacji iterowalnego obiektu. Jeśli używasz Kolekcji - po prostu uzyskaj rozmiar taki, jaki zaproponował Kazekage Gaara - zazwyczaj jest to najlepsze podejście pod względem wydajności.

Dlaczego po prostu nie użyjesz size()metody na swoim, Collectionaby uzyskać liczbę elementów?

Iterator służy tylko do iteracji, nic więcej.

Zamiast używać pętli i zliczać każdy element lub korzystać z biblioteki innej firmy, możemy po prostu wpisać iterowalną iterację w ArrayList i uzyskać jej rozmiar.

((ArrayList) iterable).size();