[python/inflearn/section2] 대표값

본 내용은 파이썬 알고리즘 문제풀이 (코딩테스트 대비) 강의를 토대로 작성하였습니다.

이미지 클릭시, 강의로 이동됩니닷 :)

# 대표값
import sys
import math
#sys.stdin=open("input.txt","rt")

n = int(input())
num_list = list(map(int, input().split()))
total = sum(num_list)
avg = int(total/n+0.5)
tmp = 0
res = 999
find_idx = 0
for idx, val in enumerate(num_list):
    # 평균과 가장 가까운 학생은 몇번쨰
    # 차이가 적을수록 평균과 가까움
    tmp = abs(val - avg) # 차이를 저장한다.
    if tmp < res :
        res = tmp
        score = val
        find_idx = idx
    elif tmp == res :
        # 차이는 같은데 값이 더 크다면 바꿔준다.
        if score < val :
            score = val
            find_idx = idx

print(avg, find_idx+1)