[백준]2108번 통계학 - 정렬

문제 설명

import sys
from collections import Counter

n = int(sys.stdin.readline())
a = []
for i in range(n):
    a.append(int(sys.stdin.readline()))
a.sort()

def avg():
    print(round(sum(a)/len(a)))

def center():    
    print(a[(len(a)//2)])

def max():
    result = Counter(a)
    result_ = result.most_common()
    if len(a) > 1:
        if result_[0][1] == result_[1][1]:
            print(result_[1][0])
        else:
            print(result_[0][0])
    else:
        print(result_[0][0])

def ran():
    print(a[len(a) - 1] - a[0])

avg()
center()
max()
ran()

 

코드 설명

먼저 입력되는 값들을 전부 리스트에 넣어두고 정렬된 상태로 진행한다. 그 다음 문제에서 요구하는 기능을 전부 함수로 만들어 처리하였다. 

avg() 함수 : 평균 값을 구하기 위해 리스트를 sum() 함수로서 전부 더한 값을 리스트의 길이로 나누어 구현하였다.

center() 함수 : 이미 리스트가 정렬되 있으므로 중앙의 값을 구하기 위해, 리스트의 길이에서 //2 를 통해 2로 나누고 소수점을 버리고
                      그 값을 인덱스로 이용하여 구현하였다.

max() 함수 : colletions 모듈을 import 하여 Counter 클래스를 이용하였다. Counter(a)를 통하여 리스트의 원소 값을 카운트 하고 
                   딕셔너리 형식으로 return 받는다. 이를 most_common() 함수를 통하여 수가 가장 많이 등장한 순서대로 정렬된 리스트를 

                   return 받아 다시 저장한다.

                   만약 리스트의 길이가 1 이하라면 리스트의 첫번째 값을 출력하고 그렇지 않다면, 최대 빈도수가 같은 경우를 고려하여 같을
                   경우 두번째 수를 출력하도록 구현하였다. 

ran() 함수 : 정렬된 리스트에서 가장 큰값은 리스트의 길이에서-1 인덱스에 존재하고 (a[-1]로도 접근이 가능), 가장 작은 값은 0 인덱스에

                 존재하기 때문에 위와 같이 구현하였다.

 

Point : 산술평균, 중앙, 범위는 간단한 연산을 통하여 쉽게 구현할 수 있다. 하지만 최빈값 같은 경우에는 따로 colletions 모듈의 클래스와 메소드를 이용하여 구현하는 것이 간편하다. Counter() 함수를 통하여 빈도 수를 구하고 most_common() 함수를 통하여 최빈 값을 정렬된 상태로 받아 처리하는 방법을 알아야 문제를 해결할 수 있다.