Biraz konu dışı olabilir, ama bu bizim için List<T>değilStream<T> .
İlk önce bir takeutil yöntemine sahip olmanız gerekir . Bu yöntemler ilk nunsurları alır :
static <T> List<T> take(List<T> l, int n) {
if (n <= 0) {
return newArrayList();
} else {
int takeTo = Math.min(Math.max(n, 0), l.size());
return l.subList(0, takeTo);
}
}
sadece gibi çalışıyor scala.List.take
assertEquals(newArrayList(1, 2, 3), take(newArrayList(1, 2, 3, 4, 5), 3));
assertEquals(newArrayList(1, 2, 3), take(newArrayList(1, 2, 3), 5));
assertEquals(newArrayList(), take(newArrayList(1, 2, 3), -1));
assertEquals(newArrayList(), take(newArrayList(1, 2, 3), 0));
şimdi takeWhiledayalı bir yöntem yazmak oldukça basit olacaktake
static <T> List<T> takeWhile(List<T> l, Predicate<T> p) {
return l.stream().
filter(p.negate()).findFirst(). // find first element when p is false
map(l::indexOf). // find the index of that element
map(i -> take(l, i)). // take up to the index
orElse(l); // return full list if p is true for all elements
}
şu şekilde çalışır:
assertEquals(newArrayList(1, 2, 3), takeWhile(newArrayList(1, 2, 3, 4, 3, 2, 1), i -> i < 4));
bu uygulama listeyi birkaç kez tekrarlar, ancak ekleme O(n^2)işlemi eklemez . Umarım bu kabul edilebilir.