[프로그래머스/Python] 모의고사

수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다.

수포자는 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, ...

1번 문제부터 마지막 문제까지의 정답이 순서대로 들은 배열 answers가 주어졌을 때, 가장 많은 문제를 맞힌 사람이

누구인지 배열에 담아 return 하도록 solution 함수를 작성해주세요.

 

---

• 멍청한 나의 접근법(딕셔너리를 사용)

def solution(answers):
    first=[1,2,3,4,5]
    second=[2,1,2,3,2,4,2,5,2,1,2,3,2,4,2,5,2,1,2,3,2,4,2,5]
    third=[3,3,1,1,2,2,4,4,5,5,3,3,1,1,2,2,4,4,5,5]
    
    count=0 
    ans_dic={'first':0, 'second':0, 'third':0 }
    
    for ans in answers:
        if first[count%len(first)]==ans:ans_dic['first']+=1
        if second[count%len(second)]==ans:ans_dic['second']+=1 
        if third[count%len(third)]==ans:ans_dic['third']+=1 
        count+=1
    
    ans_dic=sorted(ans_dic, reverse=True)

    flag=1
    answer = []
    
    for pair in ans_dic: 
        if flag==1:

            if pair[0]=='first': answer.append(1)
            elif pair[0]=='second':answer.append(2)
            elif pair[0]=='third':answer.append(3)
            tmp=pair[1]
            flag+=1 
            
        elif tmp==pair[1]:
            if pair[0]=='first': answer.append(1)
            elif pair[0]=='second':answer.append(2)
            elif pair[0]=='third':answer.append(3)

        
    return answer

멍청 포인트 

1. 필요 한 정보가 수학을 포기한 사람의 정보, 해당 사람이 맞춘 점수라고 생각해서 dict 타입이어야 한다고 생각함. 
2. 딕셔너리 타입은 정렬 이후에 리스트를 반환한다는 것을 몰랐다(...두배로 복잡해짐)

---

• 스마트 율무의 접근법

def solution(answers):
    first=[1,2,3,4,5]
    second=[2,1,2,3,2,4,2,5]
    third=[3,3,1,1,2,2,4,4,5,5]
    count=0
    
    ans_max = [0,0,0]
    for ans in answers:
        if first[count%len(first)]==ans:ans_max[0]+=1
        if second[count%len(second)]==ans:ans_max[1]+=1
        if third[count%len(third)]==ans:ans_max[2]+=1 
        count+=1
        
    answer=list(filter(lambda x: ans_max[x]==max(ans_max),range(len(ans_max))))
    
    answer=[i+1 for i in answer]
    

    return answer

풀이 포인트

1. 각각 수포자1, 수포자2, 수포자3의 배열(답안)이 answer 배열의 길이와 상관없이 반복되어야하므로, 인덱스를 각 답안 배열의 길이에 맞게 나눈 나머지를 인덱스로 사용해야한다.
   (5일 경우 0 -> 1 -> 2 -> 3 -> 4 -> 5 -> 0,,,를 반복)
2. 리스트를 사용한다. 각 원소의 인덱스는 사람의 번호(ex수포자1)를 의미하고, 해당 원소는 맞춘 갯수가 되게 된다.
   (ex/ ans_max 리스트)
3. 해당 리스트 중에서 가장 높은 값을 가진 인덱스를 추출하는 과정을 filter로 거친다. 
   filter(lambda x: ans_max[x]==max(ans_max), range(len(ans_max))) 
   - lambda 함수를 사용해 ans_max 리스트에서 최대값을 구하는 식을 생성한다.
   - filter 함수를 사용해 사용할 메소드(lambda)를 입력하고, 리스트(ans_max) 길이만큼 돌아간다.