def solution(answers):
answer = []
a = [1, 2, 3, 4, 5]
b = [2, 1, 2, 3, 2, 4, 2, 5]
c = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
a1=0;b1=0;c1=0
n = len(answers)
for i in range(n):
if(a[i%5]==answers[i]): a1+=1
if(b[i%8]==answers[i]): b1+=1
if(c[i%10]==answers[i]): c1+=1
if(a1>b1 and a1>c1): answer.append(1)
elif(b1>a1 and b1>c1): answer.append(2)
elif(c1>a1 and c1>b1): answer.append(3)
elif(a1==b1 and b1>c1): answer.append(1); answer.append(2)
elif(a1==c1 and c1>b1): answer.append(1); answer.append(3)
elif(c1==b1 and b1>a1): answer.append(2); answer.append(3)
elif(a1==b1 and b1==c1): answer.append(1); answer.append(2); answer.append(3)
return answer
댓글