[프로그래머스 LV 1] 성격 유형 검사하기 (파이썬/python)

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

프로그래머스

 

 

단순 구현문제 친구기다리다가 하나 품

 

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

    character_list = ["R", "T", "C", "F", "J", "M", "A", "N"]

    character_score = {
        "R": 0,
        "T": 0,
        "C": 0,
        "F": 0,
        "J": 0,
        "M": 0,
        "A": 0,
        "N": 0,
    }

    query_score = {
        "1": 3,
        "2": 2,
        "3": 1,
        "4": 0,
        "5": 1,
        "6": 2,
        "7": 3
    }

    for i in range(len(survey)):
        choice = int(choices[i])
        first_character = survey[i][0]
        second_character = survey[i][1]

        if choice <= 3:
            character_score[first_character] += query_score[str(choice)]

        if choice >= 5:
            character_score[second_character] += query_score[str(choice)]

    for i in range(0, 8, 2):
        first_score = character_score[character_list[i]]
        second_score = character_score[character_list[i + 1]]

        if first_score >= second_score:
            answer += character_list[i]

        else:
            answer += character_list[i + 1]

    return answer