프로그래머스 lv1 - 평균 구하기

간단하게 구현해봤습니다.

 

class Solution {
    public double solution(int[] arr) {
        double answer = 0;

        for(int i=0; i<arr.length; i++){
            answer += arr[i];
        }
        return answer/=arr.length;
    }
}

 

 

 

 

 

아래는 다른사람의 풀이입니다.

스트림을 활용하니 코드가 굉장히 깔끔하게 변해서 마음에 들었습니다.

• Arrays.stream(array).average().orElse(0);: 배열을 스트림으로 변환하고, 평균값을 계산합니다. 만약 배열이 비어있다면 orElse(0)에 의해 0을 반환한다고 합니다. 또한 소수점 값까지 표현을 위해서 앞의 return을 Arrays.stream(array).average().orElse(0.0);와 같이 수정하고 반환타입도 double로 변경하면 됩니다.

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));
    }
}

 

위의 코드를 프로그래머스에서 입력할 때는 다음과 같이 수정하면 됩니다!

import java.util.Arrays;
class Solution {  
        public double solution(int[] arr) {
      return Arrays.stream(arr).average().orElse(0.0);
    }
}