[BaekJoon] 2108번 : 통계학 (Python)

코딩 1일 1문제! 오늘의 문제는 백준의 통계학 입니다.

2108번: 통계학

Solution

from collections import Counter

numbers = []
for _ in range(int(input())):
    num = int(input())
    numbers.append(num)

numbers.sort()

cnt = Counter(numbers).most_common(2)

print(round(sum(numbers) / len(numbers)))
print(numbers[len(numbers) // 2])
if len(numbers) > 1:
    if cnt[0][1] == cnt[1][1]:
        print(cnt[1][0])
    else:
        print(cnt[0][0])
else:
    print(cnt[0][0])
print(max(numbers) - min(numbers))

Solution 풀이

먼저 몇 개의 숫자를 입력받을 것인지 입력받은 다음

그 수만큼의 숫자를 입력 받아 numbers에 append 시켜줍니다.

그 다음 numbers를 오름차순으로 정렬해둡니다.

 

먼저 산술 평균입니다.

산술평균은 numbers의 합 sum(numbers)를 numbers의 길이 len(numbers) 로 나누어준 값을

소수점 첫째자리를 기반으로 반올림하기 위하여 round를 활용하여 반올림 합니다.

round(sum(numbers) / len(numbers))

중앙값은

숫자의 개수가 홀수이므로 중앙 index인 len(numbers) // 2를 활용하여 숫자를 가져와 출력합니다.

numbers[len(numbers) // 2]

최빈값은

가장 빈도수가 높은 숫자이므로 각 숫자가 얼마나! 등장하는지 collection의 Counter를 활용하여 빈도수를 구합니다.

그 다음 최빈값이 여러개일때 두번째로 작은 값을 출력하기 위해서

most_common(2)를 활용하여 빈도수가 높은 숫자 2개를 가져옵니다.

만약 numbers의 길이가 2이상이면 두번째로 작은값을 출력하고 그렇지 않은 경우에는!

가장 빈도수가 높은 값을 그대로 출력합니다.

cnt = Counter(numbers).most_common(2)

if len(numbers) > 1:
    if cnt[0][1] == cnt[1][1]:
        print(cnt[1][0])
    else:
        print(cnt[0][0])
else:
    print(cnt[0][0])

마지막으로 범위는 최댓값 - 최솟값 이므로 max(numbers)에서 min(numbers)를 뺀 값을 출력하면됩니다.

print(max(numbers) - min(numbers))

 

이렇게 다풀고!

제출을 했는데!!!!!!

롸...? 시간초과가 났습니다.

 

혹시....? 하는 마음에 pypy3로 바꾸어 실행하니

통과하였습니다.

 

그럼 Python3에서도 시간초과가 안나려면!

int(input())

숫자를 입력받을 때 위처럼 input()을 활용하던 것을

import sys

int(sys.stdin.readline())

sys의 stdin.readline()을 활용하면!

위처럼 통과가 되는 것을 볼 수 있습니다.

👨🏻‍💻 코드 ( Solution ) - SOMJANG 20230321

from collections import Counter
import sys

def statistics(numbers):
    numbers.sort()    
    
    cnt = Counter(numbers).most_common(2)
    
    print(round(sum(numbers) / len(numbers)))
    print(numbers[len(numbers) // 2])
    
    if len(cnt) > 1:
        if cnt[0][1] == cnt[1][1]:
            print(cnt[1][0])
        else:
            print(cnt[0][0])
    else:
        print(cnt[0][0])
    print(max(numbers) - min(numbers))
    
    
if __name__ == "__main__":
    numbers = []
    for _ in range(int(sys.stdin.readline())):
        num = int(sys.stdin.readline())
        numbers.append(num)
        
    statistics(numbers=numbers)

 

읽어주셔서 감사합니다.

SOMJANG/CODINGTEST_PRACTICE