Question

今天在解决HackerRank上的this问题时，我使用了Array stream .sum（）函数对所有条目求和并继续我的算法。但总而言之，我发现我的算法在某些情况下失败了。我使用diff来发现它通过了99％的情况，1％的输出几乎相等，但小于原始答案。这就是为什么我用for循环替换了流.sum（），并且意外地它通过了所有的测试用例。我试过但无法确定这种不确定的行为。

我使用stream.sum（）的实现：

public class MandragoraForest {

    public static void main(String[] args) {
        InputReader in = new InputReader(System.in);
        for (int i = in.nextInt(); i > 0; i--) {
            int number = in.nextInt();
            int[] h = new int[number];
            for (int j = 0; j < number; j++) h[j] = in.nextInt();
            System.out.println(new MandragoraForestSolver().solve(h));
        }
    }
}

class MandragoraForestSolver {

    public long solve(int[] h) {
        if (h.length==1) return h[0];
        Arrays.parallelSort(h);
        long sum = Arrays.stream(h)
                .sum();
        long ans = -1;

        for (long i=0, strength = 2; i<h.length; i++, strength++) {
            sum -= h[(int)i];
            ans = Math.max(ans, strength * sum);
        }
        return ans;
    }
}

没有Java流的实现：

public class MandragoraForest {

    public static void main(String[] args) {
        InputReader in = new InputReader(System.in);
        for (int i = in.nextInt(); i > 0; i--) {
            int number = in.nextInt();
            int[] h = new int[number];
            long sum = 0;
            for (int j = 0; j < number; j++) {
                h[j] = in.nextInt();
                sum += h[j];
            }
            System.out.println(new MandragoraForestSolver().solve(h, sum));
        }
    }
}

class MandragoraForestSolver {

    public long solve(int[] h, long sum) {
        if (h.length==1) return h[0];
        Arrays.parallelSort(h);


        long ans = -1;

        for (long i=0, strength = 2; i<h.length; i++, strength++) {
            sum -= h[(int)i];
            ans = Math.max(ans, strength * sum);
        }
        return ans;
    }
}

有什么东西我错过了吗？这种行为可能是什么原因？

Answer 1

使用流和循环之间存在一个显着差异 - 算术溢出的可能性。

Arrays.stream(int[])会返回IntStream，其sum()方法会返回int个结果。如果总和超过Integer.MAX_VALUE，则会发生静默整数溢出。

但是，您的循环总和是将int值添加到long总计，这不会受到算术溢出的影响。

其中一个测试中的整数总和必须超过Integer.MAX_VALUE，测试long用于（正确）计算总数。

如果您想使用流汇总，则需要将IntStream转换为LongStream，您可以这样做：

long sum = Arrays.stream(big).asLongStream().sum();

Java 8 Stream.sum（）的错误行为

1 个答案: