[Programmers/Go] 평균 구하기

1. 문제

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

제한사항

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

2. 풀이

// 평균 구하기
package average

func Solution(arr []int) float64 {
	sum := 0

	for _, value := range arr {
		sum += value
	}

	return float64(sum) / float64(len(arr))
}

3. 테스트

package average

import "testing"

func TestAverage(t *testing.T) {
	input := [][]int{{1, 2, 3, 4}, {5, 5}}
	expect := []float64{2.5, 5}

	for i := range input {
		result := Solution(input[i])
		if result != expect[i] {
			t.Errorf("Test%d: Wrong result", i+1)
			t.Error("expect:", expect[i])
			t.Error("result:", result)
		}
	}
}

programmers.co.kr/learn/courses/30/lessons/12944

코딩테스트 연습 - 평균 구하기