[프로그래머스] 평균 구하기

나의 풀이

import java.util.Arrays;

class Solution {
    public double solution(int[] arr) {
        return Arrays.stream(arr).average().getAsDouble();
    }
}

이번에는 stream을 사용해서 문제를 해결하고자 했다.

익숙한 for문, while문에서 벗어나고 싶었다.

 

다른 사람의 풀이

import java.util.Arrays;

public class GetMean {
    public int getMean(int[] array) {
      return (int) Arrays.stream(array).average().orElse(0);
    }

    public static void main(String[] args) {
        int x[] = {5, 4, 3};
        GetMean getMean = new GetMean();
        // 아래는 테스트로 출력해 보기 위한 코드입니다.
        System.out.println("평균값 : " + getMean.getMean(x));
    }
}

출처 : https://school.programmers.co.kr/learn/courses/30/lessons/12944/solution_groups?language=java

 

• .orElse(0): 평균값이 없을 경우(빈 배열) 0을 반환
• (int): 평균은 double인데, 정수로 강제 형변환(casting)합니다. 예: 4.5 → 4