Algorithm프로그래머스

[프로그래머스] 코딩테스트 풀이 - 모의고사

ALGORITHM, BRUTE FORCE, CODING TEST, PYTHON, 완전탐색

Reference : 프로그래머스 코딩 테스트 연습

문제 설명

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

입출력 예

answers return
[1,2,3,4,5] [1]
[1,3,2,4,2] [1,2,3]

조건 분석

  • 나와야 하는 결과

    1
    2
    가장 높은 점수를 받은 [사람 번호]
    동점일경우 오름차순 정렬

  • 문제를 풀기위해 주어진 데이터

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    # 문제 해결에 필요한 데이터

    1. 3명의 수포자의 찍는 방식
     - 1번 : 1~5까지 반복
     - 2번 : 홀수번째는 2, 짝수번째는 2를 제외한 1~5 반복
     - 3번 : 3,1,2,4,5를 각 2번씩 반복

    2. 정답 데이터

    # 제한조건

    1. 최대 문제 갯수(1만문제)
    2. 문제 정답 제한 (1~5)
    3. 동점자가 있을경우 오름차순 정렬

  • 간단 알고리즘

    1
    1. 정답마다 각 인원의 배열을 비교한다

풀이

사용언어 : Python 3

answers 리스트를 수포자의 답과 비교해야되는데 answers의 값이 무작위로 들어있으므로 완전탐색으로 풀어야 한다.

완전탐색이 카테고리인 문제를 푸는 방법은

  • Brute Force (무작위 대입)
  • DFS (깊이 우선 탐색)
  • BFS (너비 우선 탐색)
  • 백트래킹

정도가 있다. 자세한건 타 포스팅 참조

변수자체가 한개밖에 없으므로 단순하게 무작위대입으로 푸는것이 좋다.

1~3번까지의 반복되는 숫자의 패턴자체가 정해져있으므로 먼저 패턴을 정의한다.

1
2
3
4
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]

그 뒤는 알고리즘에 따라 변수를 정의한다.

  1. answers[0]과 parttern1~3의 첫번째 배열을 비교한다.
  2. 같을경우 각 패턴에 1점씩 부여한다.
  3. 위 과정을 반복한다.
  4. 끝나고 가장 높은 배열을 출력한다.
1
2
3
# 각 패턴별 점수판 변수 선언
    score = [0, 0, 0]
    result = []
1
2
3
4
5
6
7
8
9
# answers[0]과 parttern1~3의 첫번째 배열을 비교한다.
    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
1
2
3
4
5
6

for idx, s in enumerate(score):
    if s == max(score): # 최대값 결과값 배열에 넣는다.
        result.append(idx+1)

return result
공유하기