파이썬 코딩테스트 스터디 2주차 #5 - 문제 04 모의고사 - 수포자

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

프로그래머스

 

코딩테스트 연습 > 완전탐색 > 모의고사

수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려고한다.

- 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...

1번 문제부터 마지막 문제까지의 정답이 순서대로 저장된배열 answers가 주어졌을 때,
가장 많은 문제를 맞힌 사람이 누구인지배열에 담아 반환하도록 solution()함수를 작성하시오.

제약조건
1) 시험은 최대 10,000문제로 구성되어있다.
2) 문제의 정답은 1,2,3,4,5 중 하나
3) 가장 높은 점수를 받은 사람이 여럿이라면 반환하는 값을 오름차순으로 정렬

 

 

아래 #[중요!] 부분에서 i % len 부분이 조금 헷갈린다.

이 부분 차근차근이 손으로 그려가면서 보는 것도 추천!

def solution(answers):
    answer = []
    
    #각 수포자1,2,3의 정답 갯수를 위한 배열 선언 및 초기화
    cnt=[0]*3
    
    # answer 길이만큼 index돌면서 계산하기 위해 
    for i ,vali in enumerate(answers):
    	# 수포자가 3명으로 3번 반복
        for j, valj in enumerate(patterns):
        	# [중요!] 
            # answer의 길이가 각 수포자의 답변보다 긴 경우의 수를 위해
            # i%len(valj)라고 했는데 
            # i < len(valj) 는 무조건 i % len(valj) = 그냥 i다 
            # i = len(valj) 즉, item 마지막 index가 수포자 마지막 index를 넘어갈때
            # i % len(valj) = 0 이 되면서 다시 수포자의 처음 0번째 index와 비교
            if vali == valj[i%len(valj)]:
                cnt[j]+=1
    
    max_cnt = max(cnt)

    for i in range(len(cnt)):
        if max_cnt == cnt[i]:
            answer.append(i+1)

    answer.sort()
    return answer

#수포자 답변 선언
patterns =[
    [1, 2, 3, 4, 5],
    [2, 1, 2, 3, 2, 4, 2, 5],
    [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]    
]

 

ㅇ

다른 풀이

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