본문 바로가기

Algorithm & SQL/Baekjoon

[Algorithm] [Python] 백준/BOJ - 2822 _ 점수계산

2822 - 점수 계산


INDEX


문제 설명


상근이는 퀴즈쇼의 PD이다. 이 퀴즈쇼의 참가자는 총 8개 문제를 푼다. 참가자는 각 문제를 풀고, 그 문제를 풀었을 때 얻는 점수는 문제를 풀기 시작한 시간부터 경과한 시간과 난이도로 결정한다. 문제를 풀지 못한 경우에는 0점을 받는다. 참가자의 총 점수는 가장 높은 점수 5개의 합이다.

상근이는 잠시 여자친구와 전화 통화를 하느라 참가자의 점수를 계산하지 않고 있었다. 참가자의 8개 문제 점수가 주어졌을 때, 총 점수를 구하는 프로그램을 작성하시오.

입력


8개 줄에 걸쳐서 각 문제에 대한 참가자의 점수가 주어진다. 점수는 0보다 크거나 같고, 150보다 작거나 같다. 모든 문제에 대한 점수는 서로 다르다. 입력으로 주어지는 순서대로 1번 문제, 2번 문제, ... 8번 문제이다.

출력


첫째 줄에 참가자의 총점을 출력한다. 둘째 줄에는 어떤 문제가 최종 점수에 포함되는지를 공백으로 구분하여 출력한다. 출력은 문제 번호가 증가하는 순서이어야 한다.

에제 입출력


input : 20
30
50
48
33
66
0
64

output : 261
3 4 5 6 8

제출 코드


from sys import stdin

arr = []
result = []
total = 0
for i in range(8):
    arr.append(int(stdin.readline()))

for i in range(5):
    total += max(arr)
    idx = arr.index(max(arr))
    result.append(idx+1)
    arr[idx] = 0

print(total)
print(*(sorted(result)))

코드 설명


문제는 정말 매우 간단하다.

입력을 받은 뒤, 입력값이 들어있는 배열에서 최대값 5개와 해당 원소들의 idx값만 알아내면 된다.

따라서, max()index()를 이용하였다.

max()를 이용하여 배열 내 최대값을 찾고, 해당 배열의 원소 위치를 찾아낸 뒤 이를 result 배열에 추가한다.

다음으로, 해당 배열의 길이를 건드리지 않으면서 최대값을 중복되어 찾지 않도록 하기 위해 원소값을 0으로 변경시켜줬다.