"""
입출력 예시)
answers = [1,2,3,4,5] -> [1]
answers = [1,3,2,4,2] -> [1,2,3]
"""
def solution(answers):
stu_1 = [1, 2, 3, 4, 5] # 1번 수포자가 찍는 방식
stu_2 = [2, 1, 2, 3, 2, 4, 2, 5] # 2번 수포자가 찍는 방식
stu_3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5] # 3번 수포자가 찍는 방식
score = [0, 0, 0] # 각 수포자에 대한 점수
answer = [] # 가장 많은 문제를 맞힌 사람
for idx, ans in enumerate(answers):
if ans == stu_1[idx%len(stu_1)]: # idx번째 정답이 1번 수포자의 idx번째 답과 같다면
score[0] += 1
if ans == stu_2[idx%len(stu_2)]: # idx번째 정답이 2번 수포자의 idx번째 답과 같다면
score[1] += 1
if ans == stu_3[idx%len(stu_3)]: # idx번째 정답이 3번 수포자의 idx번째 답과 같다면
score[2] += 1
for idx, s in enumerate(score):
if s == max(score):
answer.append(idx+1) # idx는 0~2까지인데 수포자 번호는 1~3까지라서 idx+1
return answer
enumerate()-> 인덱스와 원소를 동시에 접근하면서 루프 돌리기 가능
# enumerate()
answers = [1,3,2,4,2]
for idx, ans in enumerate(answers):
print(idx, ans)
"""
idx ans
0 1
1 3
2 2
3 4
4 2
"""