[Level 1] 모의고사

문제 : https://programmers.co.kr/learn/courses/30/lessons/42840

코딩테스트 연습 - 모의고사

▶ 코드

def solution(answers):
    # 1번 사람 : 1/2/3/4/5         -> cycle: 5개
    # 2번 사람 : 21/23/24/25       -> cycle: 8개
    # 3번 사람 : 33/11/22/44/55    -> cycle: 10개
    
    p1 = [1,2,3,4,5]
    p2 = [2,1,2,3,2,4,2,5]
    p3 = [3,3,1,1,2,2,4,4,5,5]

    c1 = 0
    c2 = 0
    c3 = 0

    # 길이가 다른 두 배열의 각 원소 비교 (answers는 가변 길이)(p1,p2,p3 는 불변 길이)
    for i in range(len(answers)):
        if p1[i%len(p1)] == answers[i]:
            c1 += 1
        if p2[i%len(p2)] == answers[i]:
            c2 += 1
        if p3[i%len(p3)] == answers[i]:
            c3 += 1
    #print(c1,c2,c3)

    max_value = max(c1,c2,c3)
    res = []

    if c1 >= max_value:
        res.append(1)
    if c2 >= max_value:
        res.append(2)
    if c3 >= max_value:
        res.append(3)

    res.sort()
    return res

    
#solution([1,2,3,4,5])
#solution([1,3,2,4,2])

 

▶ 매개변수로 들어오는 answers의 길이만큼 p1,p2,p3 배열 만들어주면 시간초과 

▶ 길이가 다른 두 배열의 각 원소 비교하는 방법 으로 풀이해야됨