[BackJoon]15596번 스트림으로 풀기

문제

정수 n개가 주어졌을 때, n개의 합을 구하는 함수를 작성하시오.

작성해야 하는 함수는 다음과 같다.

Java: long sum(int[] a); (클래스 이름: Test)

a: 합을 구해야 하는 정수 n개가 저장되어 있는 배열 (0 ≤ a[i] ≤ 1,000,000, 1 ≤ n ≤ 3,000,000)

리턴값: a에 포함되어 있는 정수 n개의 합

 

풀이

 

함수 안에 스트림을 집어넣어 구현해 보았다.

 

정확하게 이유는 모르겠지만 중간에 mapToLong()을 제거하면 틀린다. 숫자 범위 때문인가.

 

아무튼 간략한 풀이!

 

import java.util.Arrays;

public class Prob15596Stream {
    long sum(int[] a) {
        long ans = Arrays.stream(a)
                .mapToLong(Long::valueOf)
                .sum();
        return ans;
    }
}