2022 카카오 성격 유형 검사하기 [파이썬]

📌 문제

성격 유형 검사하기 문제 링크

📌 코드

def solution(survey, choices):
    
    result = ''
    points = [3, 2, 1, 0, 1, 2, 3]
    caseList = ['R', 'T', 'C', 'F', 'J', 'M', 'A', 'N']
    caseDict = dict(zip(caseList, [0 for _ in range(len(caseList))]))
    for idx in range(len(survey)):
        first, second = survey[idx][0], survey[idx][1]
        selectPoint = choices[idx]
        if selectPoint > 4: #* 동의 관련 선택지
            caseDict[second] += points[selectPoint - 1]
        elif 0 < selectPoint < 4: #* 비동의 관련 선택지
            caseDict[first] += points[selectPoint - 1] 
    
    max_keys = []
    for i in range(0, len(caseDict), 2):
        pair = list(caseDict.items())[i:i+2]
        max_key = max(pair, key=lambda x: x[1])[0]
        max_keys.append(max_key)
    return ''.join(max_keys)