[인프런-파이썬 알고리즘 문제풀이 입문] 대표값

문제

N명의 학생의 수학성적이 주어집니다. N명의 학생들의 평균(소수 첫째자리 반올림)을 구하고, N명의 학생 중 평균에 가장 가까운 학생은 몇 번째 학생인지 출력하는 프로그램을 작성하세요. 
답이 2개일 경우 성적이 높은 학생의 번호를 출력하세요. 만약 답이 되는 점수가 여러 개일 경우 번호가 빠른 학생의 번호를 답으로 한다.

--> 평균이 74고 73, 75 순서대로 있으면 75 한 학생의 번호 출력하기

 

입력

첫줄에 자연수 N(5<=N<=100)이 주어지고, 두 번째 줄에는 N개의 자연수가 주어진다. 
학생의 번호는 앞에서부터 1로 시작해서 N까지이다.

출력

첫줄에 평균과 평균에 가장 가까운 학생의 번호를 출력한다. 
평균은 소수 첫째 자리에서 반올림합니다.

예제 입력 1 

10
65 73 66 87 92 67 55 79 75 8

예제 출력 1 

74, 9

 

풀이 

import sys
N = sys.stdin.readline()
K = list(map(int, sys.stdin.readline().split()))

avg=round(sum(K)/len(K))
temp=100000
idxlist=[]

for i in range(len(K)):
    pwr=abs(K[i]-avg)
    if (temp>=pwr):
        temp=pwr
        idxlist.append(i+1)

print(avg, max(idxlist))

 

필요한 개념

abs(숫자) - 절댓값