[파이썬 알고리즘] 대표값

대표값

📄 문제

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

▣ 입력설명
첫줄에 자연수 N(5<=N<=100)이 주어지고,
두 번째 줄에는 각 학생의 수학점수인 N개의 자연 수가 주어집니다.
학생의 번호는 앞에서부터 1로 시작해서 N까지입니다.

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

▣ 입력예제
10
45 73 66 87 92 67 75 79 75 80

▣ 출력예제
74 7

🧞 풀이

풀이

# 입력: 첫째 줄 - 학생 수 n
# 입력: 둘째 줄 - 각 학생 수학점수

# 출력: 평균 및 평균에 가장 가까운 학생의 번호 출력
# 출력: 평균은 소수 첫째 자리에서 반올림 '73.9 -> 74'

n = int(input())                                    # 정수 n 입력하여 할당
m_score = list(map(int, input().split()))           # m_score에 각 학생의 수학 점수 입력하여 리스트로 할당

average = round((sum(m_score) / len(m_score)))      # 평균 출력: sum()으로 합 구하고 m_score의 길이를 len()으로 구하여 나눔; round()로 반올림

min = abs(m_score[0] - average)                     # 최솟값 초기값 설정: 첫 번째 학생의 수학 점수 - 평균 점수
stu = 0                                             # 학생 번호 초기값 설정: 0

for i in range(1, len(m_score)):                    # 1부터 n만큼 반복; len(m_score) 대신 n을 사용해도 괜찮음
    if abs(m_score[i] - average) < min:             # i번째 학생의 수학 점수 - 평균 점수의 절대값이 최솟값보다 작을 경우
        min = abs(m_score[i] - average)             # 해당 값을 최소값으로 할당
        stu = i                                     # 학생 번호도 i로 할당

print(average, stu+1)                               # 평균과 학생 번호 출력 (stu는 인덱스 번호기 때문에 +1 해줘야 함)

 

다른 사람 풀이

# 대표값

n = int(input())
a = list(map(int, input().split()))
ave = round(sum(a)/n)
min = 2147000000
for idx, x in enumerate(a):
    tmp = abs(x-ave)
    if tmp < min:
        min = tmp
        score = x
        res = idx + 1
    elif tmp == min:
        if x > score:
            score = x
            res = idx + 1

print(ave, res)