코테 - 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 함수를 작성해주세요.

제한 조건
시험은 최대 10,000 문제로 구성되어있습니다.
문제의 정답은 1, 2, 3, 4, 5중 하나입니다.
가장 높은 점수를 받은 사람이 여럿일 경우, return하는 값을 오름차순 정렬해주세요.

---

---

접근 방법

1. 1~3번 수포자의 풀이 유형이 일정한 규칙이 있는것을 파악했다.

   1번 수포자가 찍는 방식: 1, 2, 3, 4, 5
   2번 수포자가 찍는 방식: 2, 1, 2, 3, 2, 4, 2, 5
   3번 수포자가 찍는 방식: 3, 3, 1, 1, 2, 2, 4, 4, 5, 5

2. 고민을 많이 했는데, 주로 백준에서 알고리즘을 풀다보니 푸는 방법이 달라 당황했다.

   1. 입력을 어떻게 받아..?

      입력은 정해져있는데, 많이 애먹었던게, 입력받는 매개 변수 이름이 answers 이고
      return값으로 정답을 제출하는 리스트 이름이 answer 라서 이 점에서 많은 착오가 발생했다.

3.반복문을 사용해야하나?

문제는 최대 10,000개로 이루어져 있다해서 음.. 이걸 일일히 하나씩 비교하면
' time over ' 를 만나게 될까봐 다른 방법을 고민했다.
set을 이용한 차 집합을 생각했지만 불가능하다는걸 깨닫고 다른 방법을
궁색해보다. 문제의 종류가 완전 탐색이란걸 깨닫고 반복문을 쓰기로 했다.

---

---

나의 풀이

def solution(answers):
    answer = [];
    max_value = len(answers);
    score = [0,0,0]
    # 수포자1
    student1 = [1, 2, 3, 4, 5];
    # 수포자2
    student2 = [2, 1, 2, 3, 2, 4, 2, 5];
    # 수포자3
    student3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5];

    for i in range(max_value):
        if (student1[i % 5] == answers[i]): score[0]+=1
        if (student2[i % 8] == answers[i]): score[1]+=1
        if (student3[i % 10] == answers[i]): score[2]+=1

    max_value = max(score)
    for i in range(3):
        if(score[i]==max_value): answer.append(i+1)
    return answer

추가되는 데이터를 반복자를 %연산하여 탐색하고,
최대값은 max함수를 사용해서 탐색해봤다.

더 축약한 코드

def solution(answers):
    answer = []
    students = [[1, 2, 3, 4, 5],[2, 1, 2, 3, 2, 4, 2, 5],[3, 3, 1, 1, 2, 2, 4, 4, 5, 5]]
    score = [0, 0, 0]
    for i, value in enumerate(answers):
        for j,std in enumerate(students):
            if std[i%len(std)] == value : score[j] += 1

    for i,value  in enumerate(score):
        if max(score)==value : answer.append(i+1)
    return answer


arr = [1,3,2,4,2]
print(solution(arr))

enumerate를 이용하여 i는 list의 최대값까지 탐색하며
value와 std에 각 리스트에 i값에 해당하는 index의 값을 대입하며
값을 탐색하는 코드를 만들었다.

j == for문의 반복 횟수(1,2,3)
std == 2차원 배열의 리스트
i == for문의 반복 횟수
value == answers 리스트의 각 index의 값