[백준 2108] 통계학

문제

https://www.acmicpc.net/problem/2108

2108번: 통계학

---

아이디어

각각의 함수를 정의해준다. 최빈값의 경우가 조금 까다로웠다. 

 

각각의 수에 대해 나타난 횟수를 딕셔너리로 정의했다. 각 숫자를 key, 나타난 횟수를 value로 정의했다. 

 

각각의 key, value 조합에 대해서 value가 이때가지 나타난 value의 최댓값보다 큰 경우 해당 숫자가 최빈값이라고 할 수 있기 때문에 max_ 변수를 최빈값의 빈도로 업데이트하고 리스트를 초기화한다.

 

이 때 빈도가 max_보다 크지는 않지만 같은 경우는 이 숫자도 동일하게 최빈값이기 때문에 최빈값 리스트에 추가한다. 

 

최종적으로 리스트를 정렬한 후 길이가 1이면 그대로 반환하고, 길이가 2 이상이면 2번째로 작은 수를 반환하도록 했다.

---

Solution

import sys 

T = int(input())
lst = [int(sys.stdin.readline()) for _ in range(T)]

lst.sort()

def mean(lst) :
    return round(sum(lst) / len(lst))

def median(lst) : 
    length = len(lst)
    if length % 2 == 0 : 
        return (lst[length-1] + lst[length]) / 2
    
    else : 
        return lst[length // 2]

def mode(lst) : 
    val_dict = {}
    for val in lst : 
        if val in val_dict : 
            val_dict[val] += 1 
        else : 
            val_dict[val] = 1
            
    max_ = 0
    max_lst = []
    for key, value in val_dict.items() : 
        if value > max_ :
            max_lst = []
            max_ = value
            max_lst.append(key)
            
        if value == max_  : 
            max_lst.append(key)
    
    max_lst = list(set(max_lst))
    max_lst.sort()
    
    if len(max_lst) <= 1 : 
        return max_lst[0]
        
    else : 
        return max_lst[1]
        
    

def range_(lst) : 
    max_ = max(lst)
    min_ = min(lst)
    return max_ - min_

print(mean(lst))
print(median(lst))
print(mode(lst))
print(range_(lst))