[프로그래머스]배열의 평균값

배열의 평균값 (3)

문제 설명

정수 배열 numbers가 매개변수로 주어집니다. numbers의 원소의 평균값을 return하도록 solution 함수를 완성해주세요.

---

제한사항

• 0 ≤ numbers의 원소 ≤ 1,000
• 1 ≤ numbers의 길이 ≤ 100
• 정답의 소수 부분이 .0 또는 .5인 경우만 입력으로 주어집니다.

---

입출력 예

numbers	result
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]	5.5
[89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99]	94.0

입출력 예 설명

입출력 예 #1

• numbers의 원소들의 평균 값은 5.5입니다.

입출력 예 #2

• numbers의 원소들의 평균 값은 94.0입니다.

---

나의 풀이 코드

class Solution {
    public double solution(int[] numbers) {
        double answer = 0;
        double sum = 0;
        for (int i=0; i<numbers.length; i++){
            sum = sum + numbers[i];
        }
        answer = sum/numbers.length;
        return answer;
    }
}

배운점

import java.util.Arrays;

class Solution {
    public double solution(int[] numbers) {
        return Arrays.stream(numbers).average().orElse(0);
    }
}

위에는 남이 푼 코드인데 배열을 스트림으로 바꾸고 average() 메서드를 사용하여 평균을 구한다. 이후 만약 평균이 null인 경우 0을 출력한다.

Arrays.stream 사용이 익숙해져야겠다.