Biraz konu dışı olabilir, ama bu bizim için List<T>
değilStream<T>
.
İlk önce bir take
util yöntemine sahip olmanız gerekir . Bu yöntemler ilk n
unsurları 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 takeWhile
dayalı 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.