[백준] 2108 통계학 파이썬 풀이

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

2108번: 통계학

문제

수를 처리하는 것은 통계학에서 상당히 중요한 일이다. 통계학에서 N개의 수를 대표하는 기본 통계값에는 다음과 같은 것들이 있다. 단, N은 홀수라고 가정하자.

1. 산술평균 : N개의 수들의 합을 N으로 나눈 값
2. 중앙값 : N개의 수들을 증가하는 순서로 나열했을 경우 그 중앙에 위치하는 값
3. 최빈값 : N개의 수들 중 가장 많이 나타나는 값
4. 범위 : N개의 수들 중 최댓값과 최솟값의 차이

N개의 수가 주어졌을 때, 네 가지 기본 통계값을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 수의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 단, N은 홀수이다. 그 다음 N개의 줄에는 정수들이 주어진다. 입력되는 정수의 절댓값은 4,000을 넘지 않는다.

출력

첫째 줄에는 산술평균을 출력한다. 소수점 이하 첫째 자리에서 반올림한 값을 출력한다.

둘째 줄에는 중앙값을 출력한다.

셋째 줄에는 최빈값을 출력한다. 여러 개 있을 때에는 최빈값 중 두 번째로 작은 값을 출력한다.

넷째 줄에는 범위를 출력한다.

예제 입력 1 복사

5
1
3
8
-2
2

예제 출력 1 복사

2
2
1
10

예제 입력 2 복사

1
4000

예제 출력 2 복사

4000
4000
4000
0

예제 입력 3 복사

5
-1
-2
-3
-1
-2

예제 출력 3 복사

-2
-2
-1
2

예제 입력 4 복사

3
0
0
-1

예제 출력 4 복사

0
0
0
1

 

분명 문제는 개쉬워보였는데

왜 답이 안나오는가

 

최빈값 구하는거 어렵더라고요 생각보다

무엇보다 딕셔너리 진짜 찢어버리고 싶음,,

기본기가 부족한 제탓이긴 한데

아직도 sorted랑 ,, 기본 이해가 너무 부족한 것 같아요

그리고 오늘 가장 어이없던 점

코드가 다 맞는데 계속 시간 / 메모리 초과가 떴는데요

import sys
# sys.stdin.readline()
import heapq
import copy
import itertools
from itertools import combinations
from itertools import permutations
from collections import deque
import math
sys.setrecursionlimit(100000000)
INF = 1e9
def num1(arr):
   value = round(((sum / len(arr))))
   print(value)

def num2(arr):
   print(arr[len(arr) // 2])

def num3(dic):
   x = []
   mx = max(dic.values()) # dic의 values의 max값을 추출
   for i in dic: # 이렇게 쓰면 딕셔너리의 각 키를 순회한다
    if mx==dic[i]: # 여기서 dic[i]는 i라는 키에 대응하는 value값
        x.append(i) # 순서가 key,value이므로 ,, dic[key] = value!! 

   if len(x) > 1:
      print(x[1])
   else:
      print(x[0])
   
def num4(arr):
   value = max(arr) - min(arr)
   print(value)



N = int(input())
arr = []
sum = 0
for i in range(N):
   arr.append(int(sys.stdin.readline()))
arr.sort()
dic = dict()
for elem in arr:
    if elem in dic:
     dic[elem] +=1
    else:
       dic[elem] = 1
    sum += elem


num1(arr)
num2(arr)
num3(dic)
num4(arr)

저 위에 import 많이 해두고 쓰니까 

시간초과가 뜨더라고요

그래서 ,, 인정받을 수 있는 코드는  

 

< 정답 코드>

import sys

def num1(arr):
   value = round(((sum / len(arr))))
   print(value)

def num2(arr):
   print(arr[len(arr) // 2])

def num3(dic):
   x = []
   mx = max(dic.values()) # dic의 values의 max값을 추출
   for i in dic: # 이렇게 쓰면 딕셔너리의 각 키를 순회한다
    if mx==dic[i]: # 여기서 dic[i]는 i라는 키에 대응하는 value값
        x.append(i) # 순서가 key,value이므로 ,, dic[key] = value!! 

   if len(x) > 1:
      print(x[1])
   else:
      print(x[0])
   
def num4(arr):
   value = max(arr) - min(arr)
   print(value)



N = int(input())
arr = []
sum = 0
for i in range(N):
   arr.append(int(sys.stdin.readline()))
arr.sort()
dic = dict()
for elem in arr:
    if elem in dic:
     dic[elem] +=1
    else:
       dic[elem] = 1
    sum += elem


num1(arr)
num2(arr)
num3(dic)
num4(arr)

잡다한 import를 지우면 깔끔하게 정답이 됩니다 ! ^_^ 

삽질 오지게 했지만 하나 배웠습니다ㅎㅎ,,