https://www.acmicpc.net/problem/2108
n개의 정수가 입력될 때 입력받은 정수들로 아래 4가지 연산을 수행해야 한다.
- 평균 (소수점 첫째 자리에서 반올림)
- 중앙값
- 최빈값
- 범위 (최대값 - 최소값)
최빈값을 제외한 나머지 연산을 간단하다.
각 연산을 함수로 구현했다. 각 함수는 매개변수로 입력받은 정수로 이루어진 정렬된 리스트를 받는다.
def average(nums):
return round(sum(nums) / len(nums))
def median(nums):
return nums[len(nums) // 2]
def nums_range(nums):
return nums[-1] - nums[0]핵심은 최빈값을 구하는 부분인데 이 문제를 더 어렵게 한 조건은 아래 조건이다..
- 최빈값이 여러 개라면 여러 값 중 두번째로 작은 값을 출력한다.
일단 최빈값을 구해보자.
구하는 방식은 굉장히 여러 개가 있겠지만 나는 딕셔너리를 사용했다. 처음에는 정수범위가 4000으로 좀 작길래 리스트로 만들었는데 음수까지 고려해야 하다보니 어려움이 있었다.
아래처럼 n이라는 숫자가 있을 때 n을 key로 빈도수를 value로 하는 방식으로 구현했다.
def mod(nums):
dic = {}
for i in range(len(nums)):
n = nums[i]
if n in dic:
dic[n] += 1
else:
dic[n] = 1이제 입력받은 모든 숫자들의 빈도수가 저장된 딕셔너리를 만들었으므로 최빈값을 찾고 동률에 대한 처리를 해줘야 한다.
- 현재 빈도수가
most보다 큰 경우 현재 빈도수를most로 갱신한다. most가 갱신되었다면 동률에 대한 처리를 위한 리스트를 초기화하고 현재 키 값을 리스트에 추가한다.- 현재 빈도수가
most와 같은 경우 동률에 대한 처리를 위한 리스트에 키값을 추가한다. - 반복을 마치고 동률 처리 리스트의 길이가 2 이상이라면 동률이 있다는 것이므로 오름차순 정렬해서 두번째 값을 뽑아낸다. (두번째로 작은 값 출력)
최빈값 구하기 함수 전체코드
def mod(nums):
dic = {}
for i in range(len(nums)):
n = nums[i]
if n in dic:
dic[n] += 1
else:
dic[n] = 1
most = 0
eq_list = [] # 최빈값 동률 처리를 위한 리스트
for k, v in dic.items():
if v > most: # 더 큰 값 찾았을 때
most = v
eq_list.clear()
eq_list.append(k)
elif v == most: # 현재 최빈값과 같을 때
eq_list.append(k)
eq_list = sorted(eq_list)
if len(eq_list) >= 2:
return eq_list[1]
return eq_list[0]전체코드
def average(nums):
return round(sum(nums) / len(nums))
def median(nums):
return nums[len(nums) // 2]
def nums_range(nums):
return nums[-1] - nums[0]
def mod(nums):
dic = {}
for i in range(len(nums)):
n = nums[i]
if n in dic:
dic[n] += 1
else:
dic[n] = 1
most = 0
eq_list = []
for k, v in dic.items():
if v > most: # 더 큰 값 찾았을 때
most = v
eq_list.clear()
eq_list.append(k)
elif v == most: # 현재 최빈값과 같을 때
eq_list.append(k)
if len(eq_list) >= 2:
eq_list = sorted(eq_list)
return eq_list[1]
return eq_list[0]
n = int(input())
nums = []
for _ in range(n):
nums.append(int(input()))
nums = sorted(nums)
avg = average(nums)
med = median(nums)
mod = mod(nums)
range = nums_range(nums)
print(avg)
print(med)
print(mod)
print(range)
