99클럽 코테 스터디 15일차 TIL (모의고사) - 프로그래머스

말하는 감자·2024년 8월 5일
1

99클럽 3기

목록 보기
8/42
post-thumbnail

개인 일정으로 5일차~14일차까지의 TIL을 작성 못해, 오늘부터 몰아서 작성중이다.

현재 7일차까지 완성했고, 오늘 15일차도 작성하려고 한다.

오늘 문제는 완전 탐색 알고리즘의 카테고리에 속한 문제이다.


1. 오늘의 학습 키워드

  • 완전 탐색
  • 리스트
  • 배열

2. 문제: 모의고사

문제 설명

수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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하는 값을 오름차순 정렬해주세요.

입출력 예

answersreturn
[1,2,3,4,5][1]
[1,3,2,4,2][1,2,3]

입출력 예 설명

입출력 예 #1

  • 수포자 1은 모든 문제를 맞혔습니다.
  • 수포자 2는 모든 문제를 틀렸습니다.
  • 수포자 3은 모든 문제를 틀렸습니다.

따라서 가장 문제를 많이 맞힌 사람은 수포자 1입니다.

입출력 예 #2

  • 모든 사람이 2문제씩을 맞췄습니다.

3. 나의 풀이

접근 방법

3명의 수포자가 문제를 찍은 규칙이 나와있다. 이것을 보고 문제의 정답 (answers) 리스트가 주어졌을 때, 가장 많은 문제를 맞힌 사람을 리스트 형태로 return 하는 문제이다. 우선 3명의 수포자의 규칙을 리스트로 나타내보자.

num1 = [1,2,3,4,5]
num2 = [2,1,2,3,2,4,2,5]
num3 = [3,3,1,1,2,2,4,4,5,5]

num1의 길이는 5, num2의 길이는 8, num3의 길이는 10이다. 3명의 수포자의 문제 규칙의 길이는 다 다르다. 이것을 동일하게? 문제의 정답 리스트와 비교를 하려면 수포자의 규칙을 순회해서 확인할 수 있도록 해야한다.

num2을 예시로 본다면,

index0 → 2

index1 → 1

index2 → 2

index3 → 3

index4 → 2

index5 → 4

index6 → 2

index7 → 5

입출력 예 첫번째의 answers는 [1,2,3,4,5]이다. 즉, len(answers) == 5이다. answers의 길이에 맞게 num2의 요소를 순회하려면 다음과 같이 진행하면 된다.

answers의 인덱스/ num2의 인덱스

index0 → 1 / index0 →2

index1 → 2 / index1 →1

index2 → 3 / index2 → 2

index3 → 4 / index3 → 3

index4 → 5 / index4 → 2

                / index5 → 4

                / index6 → 2

                / index7 → 5

만약 answers가 더 길다면? answers의 길이가 10, nums의 길이가 5로 다음과 같이 나타낼 수 있다.

index0/index0

index1/index1

index2/index2

index3/index3

index4/index4

index5/index0 == index(5%len(nums))

index6/index1 == index(6%len(nums))

index7/index2 == index(7%len(nums))

index8/index3 == index(8%len(nums))

index9/index4 == index(9%len(nums))

즉, answers의 현재 index에서 수포자의 길이로 나눈 나머지 값의 인덱스에 해당하는 nums를 answers와 비교하면 된다. 이것을 코드로 나타내보자.

def solution(answers): # len(answers) = 5
    num1 = [1,2,3,4,5] # len(num1) = 5
    num2 = [2,1,2,3,2,4,2,5] # len(num2) = 8
    num3 = [3,3,1,1,2,2,4,4,5,5] # len(num3) = 10
    score = [0,0,0] # 세 명의 수포자의 점수를 기록하는 리스트
    res = [] # 세 명의 수포자 중 가장 많은 문제를 맞춘 수포자를 기록하는 리스트
    
    for i, ans in enumerate(answers): # i = {0,1,2,3,4}, 완전 탐색
    # answers의 요소 인덱스를 알기 위해 enumerate()를 사용함
        if ans == num1[i%len(num1)]:
            score[0] += 1
        if ans == num2[i%len(num2)]:
            score[1] += 1
        if ans == num3[i%len(num3)]:
            score[2] += 1
    
    for i,s in enumerate(score):
        if s == max(score): 
            res.append(i+1) # i+1인 이유는 score에서 첫번째 수포자는 score에서 인덱스 값이 0이므로
        
    return res

4. 결론

오늘 푼 문제는 완전 탐색 문제로, 비교적 간단하게 해결할 수 있는 문제였다.

저번주 내내 코테를 제대로 진행하지 못했지만 다시 나아간다!!!

5. 내일 할 일

  1. 코테
    • 8, 9, 10일차 TIL 작성하기
    • 16일차 코테 풀기
    • 16일차 TIL 작성하기
    • 프로그래머스 기초 문제
    • 프로그래머스 입문 문제
    • 힙 자료구조 다시 보기
    • 그래프 기본 탐색 알고리즘 (BFS-1,2 강의)
    • 코딩도장
  2. 딥러닝
    • 딥러닝 초급 3-3~ 3-10 강의 듣기
    • 딥러닝 초급 4 강의 듣기
profile
할 수 있다

0개의 댓글