수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 1번 문제부터 마지막 문제까지 다음과 같이 찍습니다.
1번 수포자가 찍는 방식: 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, ...
2번 수포자가 찍는 방식: 2, 1, 2, 3, 2, 4, 2, 5, 2, 1, 2, 3, 2, 4, 2, 5, ...
3번 수포자가 찍는 방식: 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, ...
id_list | report |
---|---|
[1,2,3,4,5] | [1] |
[1,3,2,4,2] | [1,2,3] |
입출력 예 #1
따라서 가장 문제를 많이 맞힌 사람은 수포자 1입니다.
입출력 예 #2
import math
def solution(answers):
p1 = [1, 2, 3, 4, 5] #5
p2 = [2, 1, 2, 3, 2, 4, 2, 5] #8
p3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5] #10
p1 = (math.ceil(len(answers)/5) * p1)[:len(answers)]
p2 = (math.ceil(len(answers)/8) * p2)[:len(answers)]
p3 = (math.ceil(len(answers)/10) * p3)[:len(answers)]
count_list = [0,0,0]
for _p1, _p2, _p3, answer in zip(p1, p2, p3, answers):
if _p1 == answer:
count_list[0] += 1
if _p2 == answer:
count_list[1] += 1
if _p3 == answer:
count_list[2] += 1
return [i+1 for i, value in enumerate(count_list) if value == max(count_list)]
로직을 세우는 시간보다 문제파악에 더 오래걸린 문제입니다. 인자가 없는데 뭘 구하라는건지... 이런 생각을 처음에 가져서 시간이 생각보다 오래걸렸습니다(고정관념이 무섭네요). 보다싶이 패턴을 스스로 찾아서 로직을 세워야 되는 문제였습니다 😅
문제풀이는 다음과 같은 방식으로 진행됩니다.
math.ceil
과 slicing
을 사용해서 answer의 길이만큼 패턴 리스트를 정리해줌def solution(answers):
pattern1 = [1,2,3,4,5]
pattern2 = [2,1,2,3,2,4,2,5]
pattern3 = [3,3,1,1,2,2,4,4,5,5]
score = [0, 0, 0]
result = []
for idx, answer in enumerate(answers):
if answer == pattern1[idx%len(pattern1)]:
score[0] += 1
if answer == pattern2[idx%len(pattern2)]:
score[1] += 1
if answer == pattern3[idx%len(pattern3)]:
score[2] += 1
for idx, s in enumerate(score):
if s == max(score):
result.append(idx+1)
return result
패턴을 굳이 늘려주지 않고 인덱스를 활용해서 counting하는 방식입니다. 역시나 고정관념에 갇혀있었네요. ㅠ