Java 8嵌套循环,包含流和性能

Kon*_*ner 12 java performance nested-loops java-8 java-stream

为了练习Java 8流,我尝试将以下嵌套循环转换为Java 8流API.它计算a ^ b(a,b <100)的最大数字总和,并在我的Core i5 760上占用~0.135s.

public static int digitSum(BigInteger x)
{
    int sum = 0;
    for(char c: x.toString().toCharArray()) {sum+=Integer.valueOf(c+"");}
    return sum;
}

@Test public void solve()
    {
        int max = 0;
        for(int i=1;i<100;i++)
            for(int j=1;j<100;j++)
                max = Math.max(max,digitSum(BigInteger.valueOf(i).pow(j)));
        System.out.println(max);
    }
Run Code Online (Sandbox Code Playgroud)

我的解决方案,我希望由于并行性而更快,实际上需要0.25秒(0.19秒没有parallel()):

int max =   IntStream.range(1,100).parallel()
            .map(i -> IntStream.range(1, 100)
            .map(j->digitSum(BigInteger.valueOf(i).pow(j)))
            .max().getAsInt()).max().getAsInt();
Run Code Online (Sandbox Code Playgroud)

我的问题

  • 我做了正确的转换,还是有更好的方法将嵌套循环转换为流计算?
  • 为什么流变种比旧变种慢得多?
  • 为什么parallel()语句实际上将时间从0.19s增加到0.25s?

我知道微基准测试很脆弱,并行性只对大问题是值得的,但对于CPU来说,甚至0.1秒都是永恒的,对吗?

更新

我使用Eclipse Kepler中的Junit 4框架进行测量(它显示了执行测试所花费的时间).

我的结果为a,b <1000而不是100:

  • 传统的循环186s
  • 顺序流193s
  • 并行流55s

更新2sum+=Integer.valueOf(c+"");sum+= c - '0';(谢谢彼得!) 替换平行方法10秒,使其达到45秒.没想到这么大的性能影响!

此外,减少与CPU内核数量的并行性(在我的情况下为4)没有做太多,因为它将时间减少到44.8s(是的,它增加了a和b = 0但我认为这不会影响表现很多):

int max = IntStream.range(0, 3).parallel().
          .map(m -> IntStream.range(0,250)
          .map(i -> IntStream.range(1, 1000)
          .map(j->.digitSum(BigInteger.valueOf(250*m+i).pow(j)))
          .max().getAsInt()).max().getAsInt()).max().getAsInt();
Run Code Online (Sandbox Code Playgroud)

ass*_*ias 23

我已根据您的代码创建了一个快速而肮脏的微基准测试.结果是:

循环:3192
lambda:3140
lambda parallel:868

因此,循环和lambda是等效的,并行流显着提高了性能.由于您的基准测试方法,我怀疑您的结果不可靠.

public static void main(String[] args) {
    int sum = 0;

    //warmup
    for (int i = 0; i < 100; i++) {
        solve();
        solveLambda();
        solveLambdaParallel();
    }

    {
        long start = System.nanoTime();
        for (int i = 0; i < 100; i++) {
            sum += solve();
        }
        long end = System.nanoTime();
        System.out.println("loop: " + (end - start) / 1_000_000);
    }
    {
        long start = System.nanoTime();
        for (int i = 0; i < 100; i++) {
            sum += solveLambda();
        }
        long end = System.nanoTime();
        System.out.println("lambda: " + (end - start) / 1_000_000);
    }
    {
        long start = System.nanoTime();
        for (int i = 0; i < 100; i++) {
            sum += solveLambdaParallel();
        }
        long end = System.nanoTime();
        System.out.println("lambda parallel : " + (end - start) / 1_000_000);
    }
    System.out.println(sum);
}

public static int digitSum(BigInteger x) {
    int sum = 0;
    for (char c : x.toString().toCharArray()) {
        sum += Integer.valueOf(c + "");
    }
    return sum;
}

public static int solve() {
    int max = 0;
    for (int i = 1; i < 100; i++) {
        for (int j = 1; j < 100; j++) {
            max = Math.max(max, digitSum(BigInteger.valueOf(i).pow(j)));
        }
    }
    return max;
}

public static int solveLambda() {
    return  IntStream.range(1, 100)
            .map(i -> IntStream.range(1, 100).map(j -> digitSum(BigInteger.valueOf(i).pow(j))).max().getAsInt())
            .max().getAsInt();
}

public static int solveLambdaParallel() {
    return  IntStream.range(1, 100)
            .parallel()
            .map(i -> IntStream.range(1, 100).map(j -> digitSum(BigInteger.valueOf(i).pow(j))).max().getAsInt())
            .max().getAsInt();
}
Run Code Online (Sandbox Code Playgroud)

我也用jmh运行它,这比手动测试更可靠.结果与上述一致(每次通话微秒):

Benchmark                                Mode   Mean        Units
c.a.p.SO21968918.solve                   avgt   32367.592   us/op
c.a.p.SO21968918.solveLambda             avgt   31423.123   us/op
c.a.p.SO21968918.solveLambdaParallel     avgt   8125.600    us/op
Run Code Online (Sandbox Code Playgroud)

  • 如果你以相反的顺序运行测试,我会很高兴看到你得到了什么. (3认同)
  • @kirdie junit是可靠的,但是不要为你准备好代码.预热代码可以比第一次运行快许多倍. (3认同)
  • @PeterLawrey相同的结果(lambda parallel:836,lambda:3124,loop:3184) (2认同)