[프로그래머스/Python] Lv1. 성격 유형 검사하기

⭐ 문제 링크

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

프로그래머스

 

 

⭐ 풀이 코드

def solution(survey, choices):
    answer = ''
    
    score = {1:-3, 2:-2, 3:-1, 4:0, 5:1, 6:2, 7:3}
    types = {"R":0, "T":0, "F":0, "C":0, "M":0, "J":0, "A":0, "N":0}  # 타입별 점수
    sets = [("R","T"), ("C","F"), ("J","M"),("A","N")]  # 알파벳 순으로 입력
    
    # 점수 계산
    for i in range(len(choices)):
        if score[choices[i]] < 0:  # 부정일 경우
            types[survey[i][0]] += abs(score[choices[i]])
        elif score[choices[i]] > 0:  # 긍정일 경우
            types[survey[i][1]] += score[choices[i]]
    
    # 타입별 점수 비교
    for i in range(4):
        if types[sets[i][0]] > types[sets[i][1]]:
            answer += sets[i][0]
        elif types[sets[i][0]] < types[sets[i][1]]:
            answer += sets[i][1]
        else:
            answer += sets[i][0]
    
    return answer