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

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

프로그래머스

---

---

아이디어

1. 딕셔너리

• 성격유형은 key, 점수는 value로 사용해서 딕셔너리를 만들어서 해결했다.
• choices에 있는 결과에 맞춰 성격 유형 점수를 더해준다.

---

전체 코드

def solution(survey, choices):
    answer = ''
    N = len(survey)
    personality = {'R':0, 'T':0, 'C':0, 'F':0, 'J':0, 'M':0, 'A':0, 'N':0}
    
    for i in range(N):
        first, second = survey[i][0], survey[i][1]
        if choices[i] <= 3:
            personality[first] += 4 - choices[i]
        elif choices[i] > 4:
            personality[second] += choices[i] - 4
    
    if personality['R'] >= personality['T']:
        answer += 'R'
    else:
        answer += 'T'
    
    
    if personality['C'] >= personality['F']:
        answer += 'C'
    else:
        answer += 'F'
        
    
    if personality['J'] >= personality['M']:
        answer += 'J'
    else:
        answer += 'M'
        
    
    if personality['A'] >= personality['N']:
        answer += 'A'
    else:
        answer += 'N'
        
    print(personality)
    return answer

---

코드 설명

 

personality = {'R':0, 'T':0, 'C':0, 'F':0, 'J':0, 'M':0, 'A':0, 'N':0}

성격 유형 점수를 저장할 딕셔너리를 선언해준다.

 

    for i in range(N):
        first, second = survey[i][0], survey[i][1]
        if choices[i] <= 3:
            personality[first] += 4 - choices[i]
        elif choices[i] > 4:
            personality[second] += choices[i] - 4

문제에서 주어진 조건에 따라 점수를 더해주면 된다.