[python] programmers - 성격 유형 검사하기

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

프로그래머스

 

SOLUTION  📝

 

- 선택한 성격 유형 점수의 누적합을 저장

- 만일 성격 유형으로 선택되지 않은 값이 있다면 0으로 초기화

- 비교 세트 성격 유형에서 해당 점수가 더 높거나 혹은 같다면 사전 순으로 빠른 값을 answer에 추가

 

CODE 📌

def solution(survey, choices):
    
    # 정답 변수
    answer = ''
    # 점수를 저장할 dict
    score = {}
 
    
    def add(selection, add_score):
        
        # 만일 고른 값이 score에 있다면 add
        if selection in score:
            score[selection] += add_score
        # 아니라면 해당 값으로 지정
        else:
            score[selection] = add_score
            
            
    for i in range(len(survey)):
        # select는 점수
        select = choices[i]
        if select < 4:
            # select가 4미만이라면 앞의 성격 유형에 abs(4-select) 값을 더함
            add(survey[i][0], abs(4-select))
            # 뒤의 성격 유형이 선택되지 않는 경우 대비
            add(survey[i][1], 0) 
        elif select > 4:
            # select가 4초과라면 뒤의 성격 유형에 abs(select-4) 값을 더함
            add(survey[i][1], abs(select-4))
            # 앞의 성격 유형이 선택되지 않는 경우 대비
            add(survey[i][0], 0)
    
    
    # 성격 유형 검사로 나오지 않는 경우를 대비
    for can_choice in ("RT", "TR", "FC", "CF", "MJ", "JM", "AN", "NA"):
        add(can_choice[0], 0)
        add(can_choice[1], 0)
        
        
    if score['R'] >= score['T']:
        answer += 'R'
    else:
        answer += 'T'
    
    if score['C'] >= score['F']:
        answer += 'C'
    else:
        answer += 'F'
        
    if score['J'] >= score['M']:
        answer += 'J'
    else:
        answer += 'M'
        
    if score['A'] >= score['N']:
        answer += 'A'
    else:
        answer += 'N'
        
    return answer