Java Stream 计数操作，直到一个整数变为 0

Question

我如何在给定整数 N 的情况下使用 Java Stream，如果 N 是奇数，则从中减去 1，如果 N 是偶数，则将其除以 2，直到 N 变为 0？

这是我使用过程式风格的工作代码:

public static int solution(int num) {
    int counter = 0;
    while(num != 0) {
        num = (num % 2 == 0) ? num / 2 : num - 1;
        counter++;
    }

    return counter;
}

Answer 1

您可以使用具有相同逻辑的 IntStream.iterate:

public static long solutionStream(int num) {
    return IntStream.iterate(num, i -> i % 2 == 0 ? i / 2 : i -1)
            .takeWhile(i -> i > 0)
            .count();
}

请注意，takeWhile 仅在 Java 9+ 中可用，这里有必要结束由 iterate 产生的无限流。