丑数

/*
题目思路：
使用Java Stream来实现该功能并不常见，因为它不太适合这种需要状态和顺序的操作。
不过我们仍然可以用stream生成无穷序列，然后截取第n个丑数。
我们将使用Stream.iterate和limit来实现。
*/

import java.util.stream.Stream;
import java.util.PriorityQueue;
import java.util.HashSet;

public class UglyNumberStream {
    public int nthUglyNumber(int n) {
        PriorityQueue<Long> heap = new PriorityQueue<>();
        HashSet<Long> seen = new HashSet<>();
        heap.add(1L);
        seen.add(1L);
        long[] primes = {2, 3, 5};
        Stream<Long> stream = Stream.iterate(1L, x -> {
            long currentUgly = heap.poll();
            for (long prime : primes) {
                long newUgly = currentUgly * prime;
                if (seen.add(newUgly)) {
                    heap.add(newUgly);
                }
            }
            return heap.peek();
        }).limit(n);
        return stream.skip(n - 1).findFirst().get().intValue();
    }
}