🤖 옌지니어의 IT 기록 🤖

https://school.programmers.co.kr/learn/courses/30/lessons/118666

문제

문제 설명

나만의 카카오 성격 유형 검사지를 만들려고 합니다.
성격 유형 검사는 다음과 같은 4개 지표로 성격 유형을 구분합니다. 성격은 각 지표에서 두 유형 중 하나로 결정됩니다.

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

검사지에는 총 n개의 질문이 있고, 각 질문에는 아래와 같은 7개의 선택지가 있습니다.

• 매우 비동의
• 비동의
• 약간 비동의
• 모르겠음
• 약간 동의
• 동의
• 매우 동의

각 질문은 1가지 지표로 성격 유형 점수를 판단합니다.

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

이때 검사자가 질문에서 약간 동의 선택지를 선택할 경우 어피치형(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

입출력 예

소스 코드

# 1
def solution(survey, choices):
    answer = ''

    types = [["R","T"],["C","F"],["J","M"],["A","N"]]
    scores = [[0,0],[0,0],[0,0],[0,0]]

    # survey 기반 점수 계산
    n = len(survey)
    for idx in range(n):
        front = survey[idx][0]
        back = survey[idx][1]
        if "R" in survey[idx]:
            if choices[idx] < 4:
                scores[0][types[0].index(front)] += (4-choices[idx])
            else:
                scores[0][types[0].index(back)] += (choices[idx]-4)
        elif "C" in survey[idx]:
            if choices[idx] < 4:
                scores[1][types[1].index(front)] += (4-choices[idx])
            else:
                scores[1][types[1].index(back)] += (choices[idx]-4)
        elif "J" in survey[idx]:
            if choices[idx] < 4:
                scores[2][types[2].index(front)] += (4-choices[idx])
            else:
                scores[2][types[2].index(back)] += (choices[idx]-4)
        elif "A" in survey[idx]:
            if choices[idx] < 4:
                scores[3][types[3].index(front)] += (4-choices[idx])
            else:
                scores[3][types[3].index(back)] += (choices[idx]-4)          



    # scores를 보고 점수가 높은 유형 출력 
    for idx in range(4):
        if scores[idx][0] >= scores[idx][1]:
            answer += types[idx][0]
        else:
            answer += types[idx][1]

    return answer

풀이 # 1

• 각 타입과 점수를 2차원 배열로 생성
• survey 길이만큼 인덱스를 돌면서 점수를 계산
  • survey에서 앞글자와 뒷글자를 구분
  • 어떤 글자를 포함하고 있는지 확인하고 socres에 점수 더하기
  • 점수가 4보다 작으면 앞글자의 인덱스에 4보다 크면 뒷글자의 인덱스에 점수 더하기
• 앞에서 계산한 scores를 돌면서 점수가 높은 값의 type을 answer로 출력

# 2
def solution(survey, choices):
    answer = ''

    scores = {"A":0, "N":0, "C":0, "F":0, "M":0, "J":0, "R":0, "T":0}
    for idx, choice in enumerate(choices):
        if choice - 4 > 0:
            scores[survey[idx][1]] += choice - 4
        elif choice - 4 < 0:
            scores[survey[idx][0]] += 4 - choice

    if scores["R"] >= scores["T"]:
        answer += "R"
    else:
        answer += "T"

    if scores["C"] >= scores["F"]:
        answer += "C"
    else:
        answer += "F"

    if scores["J"] >= scores["M"]:
        answer += "J"
    else:
        answer += "M"

    if scores["A"] >= scores["N"]:
        answer += "A"
    else:
        answer += "N"
    

    return answer

풀이 #2

• 딕셔너리의 키에 유형 값에 점수를 표기
• 점수가 4보다 작으면 앞글자의 인덱스에 4보다 크면 뒷글자의 인덱스에 점수 더하기
• RT, CF, JM, AN의 점수를 각각 비교해서 answer로 출력

# 3
def solution(survey, choices):
    types = {"RT":0,"CF":0,"JM":0,"AN":0}
    
    for A,B in zip(survey,choices):
        if A not in types.keys():
            A = A[::-1]
            types[A] -= B-4
        else:
            types[A] += B-4

    answer = ''    
    for name in types.keys():
        if types[name] > 0:
            answer += name[1]
        elif types[name] < 0:
            answer += name[0]
        else:
            answer += sorted(name)[0]

    return answer

풀이 #3

• 짝이 되는 유형끼리 딕셔너리의 키로 구성
• survey에 있는 문자가 딕셔너리의 키가 아닌 경우에 키를 역순으로 바꾸고 점수 빼기 (순서 변경 O)
  
  • 점수가 높으면 (점수-4)가 0보다 크므로 양수 = 뒤에 있는 문자에 점수 빼기
  • 점수가 낮으면 (점수-4)가 0보다 작으므로 음수 = 뒤에 있는 문자에 점수 더하기
• survey에 있는 문자가 딕셔너리의 키인 경우에는 그냥 점수 더하기 (순서 변경 X)
  • 점수가 높으면 (점수-4)가 0보다 크므로 양수 = 뒤에 있는 문자에 점수 더하기
  • 점수가 낮으면 (점수-4)가 0보다 작으므로 음수 = 뒤에 있는 문자에 점수 빼기
• 위의 방식대로 점수를 더하고 빼면 점수는 결국 키의 뒷 문자 기반
  • 점수가 0보다 크면 뒷 문자를 출력하고 0보다 작은면 앞 문자를 출력

문제에 대해 설명이 자세해서 이해는 잘 되었는데 코드를 구현하려니까 좀 복잡해졌다. 첫 번째 풀이는 점수가 4보다 작으면 앞에 있는 문자에 +3, 4보다 크면 뒤에 있는 문자에 +3이 되는 것을 이용해서 풀었다. 첫 번째 풀이 기반으로 딕셔너리를 이용해서 두 번째 풀이까지 만들었는데 다른 사람 풀이를 보다 보니 세 번째 풀이가 진짜 씽크빅이어서 추가했다....문자열을 바꾸면서 점수 더하기? 도대체 이런 생각은 어떻게 하시는거에요? 해석하기도 어려웠는데...천재 아니신가요...?