프로그래머스 - 성격 유형 검사하기

def solution(survey, choices):
    answer = ''
    l = len(survey)
    mbti = ['R','T','C','F','J','M','A','N']
    result = [0]*4
    idx = 0
    
    for i in range(l):
        mul =1
        if survey[i][0]=='R' or  survey[i][0]=='T':
            if survey[i][0]=='T': mul*=-1
            idx = 0
        elif survey[i][0]=='C' or  survey[i][0]=='F':
            if survey[i][0]=='F': mul*=-1
            idx = 1
        elif survey[i][0]=='J' or  survey[i][0]=='M':
            if survey[i][0]=='M': mul*=-1
            idx = 2
        elif survey[i][0]=='A' or  survey[i][0]=='N':
            if survey[i][0]=='N': mul*=-1
            idx =3
            
        choices[i]*=-1
        choices[i] +=4
        choices[i] *= mul
        
        result[idx] +=choices[i]
        
    for i in range(4):
        if result[i]>=0:
            answer += mbti[i*2] 
        else:
            answer += mbti[i*2+1]
    
    
    return answer

 

1. mbti의 결과를 점수로 받아줄 result 생성
2. survey를 조사하여 어떤 유형을 검사하는지 확인 후 idx에 저장
3. 알파벳이 앞자리에 오는지 확인 후 T,F,M,N인 경우는 mul*=-1
4. choices결과에 -1을 곱한 후 4를 더하여 3,2,1,0,-1,-2,-3 같이 수정
5. R,C,J,A 경우에는 result값에 그대로 더하기, T,F,M,N인 경우는 mul을 곱하여 더하기
6. result값이 0이상 이면 R,C,J,A / 0보다 작으면 T,F,M,N의 결과를 나타냄