프로그래머스 – 성격 유형 검사하기 문제 Python

프로그래머스 문제풀이

문제

문제설명

나만의 카카오 성격 유형 검사지를 만들려고 합니다.

성격 유형 검사는 다음과 같은 4개 지표로 성격 유형을 구분합니다.

성격은 각 지표에서 두 유형 중 하나로 결정됩니다.

지표 번호성격 유형
1번 지표라이언형(R), 튜브형(T)
2번 지표콘형(C), 프로도형(F)
3번 지표제이지형(J), 무지형(M)
4번 지표어피치형(A), 네오형(N)

4개의 지표가 있으므로 성격 유형은 총 16(=2 x 2 x 2 x 2)가지가 나올 수 있습니다. 예를 들어, “RFMN”이나 “TCMA”와 같은 성격 유형이 있습니다.

검사지에는 총 n개의 질문이 있고, 각 질문에는 아래와 같은 7개의 선택지가 있습니다.
  • 매우 비동의
  • 비동의
  • 약간 비동의
  • 모르겠음
  • 약간 동의
  • 동의
  • 매우 동의
각 질문은 1가지 지표로 성격 유형 점수를 판단합니다.

예를 들어, 어떤 한 질문에서 4번 지표로 아래 표처럼 점수를 매길 수 있습니다.

선택지성격 유형 점수
매우 비동의네오형 3점
비동의네오형 2점
약간 비동의네오형 1점
모르겠음어떤 성격 유형도 점수를 얻지 않습니다
약간 동의어피치형 1점
동의어피치형 2점
매우 동의어피치형 3점

이때 검사자가 질문에서 약간 동의 선택지를 선택할 경우 어피치형(A) 성격 유형 1점을 받게 됩니다.

만약 검사자가 매우 비동의 선택지를 선택할 경우 네오형(N) 성격 유형 3점을 받게 됩니다.

위 예시처럼 네오형이 비동의, 어피치형이 동의인 경우만 주어지지 않고, 질문에 따라 네오형이 동의, 어피치형이 비동의인 경우도 주어질 수 있습니다.

하지만 각 선택지는 고정적인 크기의 점수를 가지고 있습니다.
  • 매우 동의나 매우 비동의 선택지를 선택하면 3점을 얻습니다.
  • 동의나 비동의 선택지를 선택하면 2점을 얻습니다.
  • 약간 동의나 약간 비동의 선택지를 선택하면 1점을 얻습니다.
  • 모르겠음 선택지를 선택하면 점수를 얻지 않습니다.

검사 결과는 모든 질문의 성격 유형 점수를 더하여 각 지표에서 더 높은 점수를 받은 성격 유형이 검사자의 성격 유형이라고 판단합니다.

단, 하나의 지표에서 각 성격 유형 점수가 같으면, 두 성격 유형 중 사전 순으로 빠른 성격 유형을 검사자의 성격 유형이라고 판단합니다.

질문마다 판단하는 지표를 담은 1차원 문자열 배열 survey와 검사자가 각 질문마다 선택한 선택지를 담은 1차원 정수 배열 choices가 매개변수로 주어집니다.

이때, 검사자의 성격 유형 검사 결과를 지표 번호 순서대로 return 하도록 solution 함수를 완성해주세요.

제한사항

  • 1 ≤ survey의 길이 ( = n) ≤ 1,000
    • survey의 원소는 "RT", "TR", "FC", "CF", "MJ", "JM", "AN", "NA" 중 하나입니다.
    • survey[i]의 첫 번째 캐릭터는 i+1번 질문의 비동의 관련 선택지를 선택하면 받는 성격 유형을 의미합니다.
    • survey[i]의 두 번째 캐릭터는 i+1번 질문의 동의 관련 선택지를 선택하면 받는 성격 유형을 의미합니다.
  • choices의 길이 = survey의 길이
    • choices[i]는 검사자가 선택한 i+1번째 질문의 선택지를 의미합니다.1 ≤ choices의 원소 ≤ 7
choices
1매우 비동의
2비동의
3약간 비동의
4모르겠음
5약간 동의
6동의
7매우 동의

입출력 예

surveychoicesresult
["AN", "CF", "MJ", "RT", "NA"][5, 3, 2, 7, 5]"TCMA"
["TR", "RT", "TR"][7, 1, 3]"RCJA"

입출력 예 설명

입출력 예 #1

1번 질문의 점수 배치는 아래 표와 같습니다.

선택지성격 유형 점수
매우 비동의어피치형 3점
비동의어피치형 2점
약간 비동의어피치형 1점
모르겠음어떤 성격 유형도 점수를 얻지 않습니다
약간 동의네오형 1점
동의네오형 2점
매우 동의네오형 3점

1번 질문에서는 지문의 예시와 다르게 비동의 관련 선택지를 선택하면 어피치형(A) 성격 유형의 점수를 얻고, 동의 관련 선택지를 선택하면 네오형(N) 성격 유형의 점수를 얻습니다.

1번 질문에서 검사자는 약간 동의 선택지를 선택했으므로 네오형(N) 성격 유형 점수 1점을 얻게 됩니다.

2번 질문의 점수 배치는 아래 표와 같습니다.

선택지성격 유형 점수
매우 비동의콘형 3점
비동의콘형 2점
약간 비동의콘형 1점
모르겠음어떤 성격 유형도 점수를 얻지 않습니다
약간 동의프로도형 1점
동의프로도형 2점
매우 동의프로도형 3점

2번 질문에서 검사자는 약간 비동의 선택지를 선택했으므로 콘형(C) 성격 유형 점수 1점을 얻게 됩니다.

3번 질문의 점수 배치는 아래 표와 같습니다.

선택지성격 유형 점수
매우 비동의무지형 3점
비동의무지형 2점
약간 비동의무지형 1점
모르겠음어떤 성격 유형도 점수를 얻지 않습니다
약간 동의제이지형 1점
동의제이지형 2점
매우 동의제이지형 3점

3번 질문에서 검사자는 비동의 선택지를 선택했으므로 무지형(M) 성격 유형 점수 2점을 얻게 됩니다.

4번 질문의 점수 배치는 아래 표와 같습니다.

선택지성격 유형 점수
매우 비동의라이언형 3점
비동의라이언형 2점
약간 비동의라이언형 1점
모르겠음어떤 성격 유형도 점수를 얻지 않습니다
약간 동의튜브형 1점
동의튜브형 2점
매우 동의튜브형 3점

4번 질문에서 검사자는 매우 동의 선택지를 선택했으므로 튜브형(T) 성격 유형 점수 3점을 얻게 됩니다.

5번 질문의 점수 배치는 아래 표와 같습니다.

선택지성격 유형 점수
매우 비동의네오형 3점
비동의네오형 2점
약간 비동의네오형 1점
모르겠음어떤 성격 유형도 점수를 얻지 않습니다
약간 동의어피치형 1점
동의어피치형 2점
매우 동의어피치형 3점

5번 질문에서 검사자는 약간 동의 선택지를 선택했으므로 어피치형(A) 성격 유형 점수 1점을 얻게 됩니다.

1번부터 5번까지 질문의 성격 유형 점수를 합치면 아래 표와 같습니다.

지표 번호성격 유형점수성격 유형점수
1번 지표라이언형(R)0튜브형(T)3
2번 지표콘형(C)1프로도형(F)0
3번 지표제이지형(J)0무지형(M)2
4번 지표어피치형(A)1네오형(N)1

각 지표에서 더 점수가 높은 T,C,M이 성격 유형입니다.

하지만, 4번 지표는 1점으로 동일한 점수입니다. 따라서, 4번 지표의 성격 유형은 사전순으로 빠른 A입니다.

따라서 "TCMA"를 return 해야 합니다.

입출력 예 #2

1번부터 3번까지 질문의 성격 유형 점수를 합치면 아래 표와 같습니다.

지표 번호성격 유형점수성격 유형점수
1번 지표라이언형(R)6튜브형(T)1
2번 지표콘형(C)0프로도형(F)0
3번 지표제이지형(J)0무지형(M)0
4번 지표어피치형(A)0네오형(N)0

1번 지표는 튜브형(T)보다 라이언형(R)의 점수가 더 높습니다.

따라서 첫 번째 지표의 성격 유형은 R입니다.

하지만, 2, 3, 4번 지표는 모두 0점으로 동일한 점수입니다.

따라서 2, 3, 4번 지표의 성격 유형은 사전순으로 빠른 CJA입니다.

따라서 "RCJA"를 return 해야 합니다.

문제해결과정

  1. 카카오에서 성격 유형 검사지를 통해서 성격 유형을 분석하고자 한다.
  2. 지표는 총 4개이며, 각 지표마다 유형은 두 가지이다.
  3. 주어진 survey와 choice 값을 통해 각 지표별 유형을 구한다.
    ex) survey = “AN” , choice = 5
    A <—1(3)—2(2)—3(1)—4(0)—5(1)—6(2)—7(3)—> N
    N을 약간동의하므로 N 유형에 +1점
  4. 두 유형중 합산값이 큰 유형을 선정한다.
  5. 합산값이 점수가 같으면, 사전 순으로 빠른 성격 유형을 선택한다.

위 같은 문제 및 제한사항을 해석하고 문제풀이를 시작하면 됩니다.

나의 풀이

우선 점수표와 지표별 점수를 합산할 Dict를 만들었다.

Survey와 choice 리스트에 대한 처리를 for문을 사용하여 합산값에 추가해주었다.

리스트 처리를 완료하고 순차적으로 각 유형의 합산값을 비교하여 성격유형을 선정해주었다.

def solution(survey, choices):
    MBTI = []
    kind = {'R':0,'T':0,'C':0,'F':0,'J':0,'M':0,'A':0,'N':0}    # 유형별 합산값을 저장할 dict
    point = {'1':3,'2':2,'3':1,'4':0,'5':1,'6':2,'7':3}         # 선택지별 점수값이 저장된 dict
    for i in range(len(survey)):                                # survey리스트의 길이만큼 반복
        value = survey[i][1] if choices[i]>4 else survey[i][0]  # choice의 값을 통해 점수를 추가할 유형을 선택
        kind[value] += point[str(choices[i])]                   # 선택된 유형에 점수 추가
    MBTI.append('T' if kind['R']<kind['T'] else 'R')            # 각 지표의 합산값 비교
    MBTI.append('F' if kind['C']<kind['F'] else 'C')
    MBTI.append('M' if kind['J']<kind['M'] else 'J')
    MBTI.append('N' if kind['A']<kind['N'] else 'A')
    return ''.join(MBTI)                                        # MBTI 리스트를 ''.join을 통해서 문자열로 변경 출력

다른 사람의 풀이

한글 변수를 사용하여 이해가 편하게 풀이과정을 서술한 풀이가 있어서 가져와봤습니다.

저는 개인적으로 한글을 사용하면 코드가 길어져서 비추하는 바입니다.

def solution(설문_조사_배열, 선택지_배열):
    지표 = {}
    지표['RT'] = 지표['TR'] = {'R': 0, 'T': 0,}
    지표['FC'] = 지표['CF'] = {'C': 0, 'F': 0,}
    지표['MJ'] = 지표['JM'] = {'J': 0, 'M': 0,}
    지표['AN'] = 지표['NA'] = {'A': 0, 'N': 0,}
    점수 = {
        '매우 비동의': 3,
        '비동의': 2,
        '약간 비동의': 1,
        '모르겠음': 0,
        '약간 동의': 1,
        '동의': 2,
        '매우 동의': 3,
    }
    비동의 = [1, 2, 3]
    동의 = [5, 6, 7]
    선택지 = {
        1: '매우 비동의',
        2: '비동의',
        3: '약간 비동의',
        4: '모르겠음',
        5: '약간 동의',
        6: '동의',
        7: '매우 동의',
    }
    answer = ''

    for 인덱스 in range(len(설문_조사_배열)):
        비동의_캐릭터, 동의_캐릭터 = 설문_조사_배열[인덱스]

        if 선택지_배열[인덱스] in 비동의:
            지표[설문_조사_배열[인덱스]][비동의_캐릭터] += 점수[선택지[선택지_배열[인덱스]]]
            continue

        if 선택지_배열[인덱스] in 동의:
            지표[설문_조사_배열[인덱스]][동의_캐릭터] += 점수[선택지[선택지_배열[인덱스]]]

    결과_배열 = [지표['RT'].items(), 지표['FC'].items(), 지표['MJ'].items(), 지표['AN'].items()]
    정렬된_배열 = []

    for 결과 in 결과_배열:
        정렬된_배열.append(sorted(결과, key=lambda x: -x[1]))

    return ''.join([캐릭터_점수_튜플[0] for [캐릭터_점수_튜플, _] in 정렬된_배열])