[프로그래머스/알고리즘] 평균 구하기 (자바/Java)

문제 설명

정수를 담고 있는 배열 arr의 평균값을 return하는 함수, solution을 완성해보세요.

 

제한사항

• arr은 길이 1 이상, 100 이하인 배열입니다.
• arr의 원소는 -10,000 이상 10,000 이하인 정수입니다.

 

Step1. 문제분석

 

우선 배열의 합을 구하고 그 배열의 길이만큼 나누면 된다.

그리고 주의할 점이 return 값이 실수로도 나온다는 점을 봐야한다.

그렇기에 double로 sum과 answer를 초기화 하고 진행한다.

 

Step2. 문제풀이 코드

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

    for(int i = 0; i < arr.length; i++){
        sum += arr[i];

    }
    answer =  sum / arr.length;
    return answer;
        
    }
}

 

이 문제를 푸는 방법은 다양했다는게 신기했다.

나는 이렇게만 풀면 된다고 생각했다. 너무 편협한 시선, 사고를 가지고 있는 것같다.

 

Step3. 다른 사람의 풀이 코드

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

 

Arrays.stream(array).average().orElse(0);

이 부분을 보면 평균값을 구하기 위해 average가 있는지도 공부를 할 필요가 있었고

stream에 대해서는 조금 더 파고들어 갈 필요가 있을 것 같다.

 

public class GetMean {
    public int getMean(int[] array) {
      if(array == null){
        return 0;
      }

      int ret = 0;
      if(array != null){
        for(int n : array){
            ret += n;
        }
      }
        return ret / array.length;
    }

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

 

이 부분은 array의 값이 0일 경우 리턴을 0 으로 해서 오류를 최소화 했다.