[Programmers] Lv.1 성격 유형 검사하기 (Go)

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

프로그래머스

 

문제 설명

나만의 카카오 성격 유형 검사지를 만들려고 합니다.
성격 유형 검사는 다음과 같은 4개 지표로 성격 유형을 구분합니다. 성격은 각 지표에서 두 유형 중 하나로 결정됩니다.

4개의 지표가 있으므로 성격 유형은 총 16(=2 x 2 x 2 x 2)가지가 나올 수 있습니다. 예를 들어, "RFMN"이나 "TCMA"와 같은 성격 유형이 있습니다.

검사지에는 총 n개의 질문이 있고, 각 질문에는 아래와 같은 7개의 선택지가 있습니다.

• 매우 비동의
• 비동의
• 약간 비동의
• 모르겠음
• 약간 동의
• 동의
• 매우 동의

각 질문은 1가지 지표로 성격 유형 점수를 판단합니다.

예를 들어, 어떤 한 질문에서 4번 지표로 아래 표처럼 점수를 매길 수 있습니다.

이때 검사자가 질문에서 약간 동의 선택지를 선택할 경우 어피치형(A) 성격 유형 1점을 받게 됩니다. 만약 검사자가 매우 비동의 선택지를 선택할 경우 네오형(N) 성격 유형 3점을 받게 됩니다.

위 예시처럼 네오형이 비동의, 어피치형이 동의인 경우만 주어지지 않고, 질문에 따라 네오형이 동의, 어피치형이 비동의인 경우도 주어질 수 있습니다.
하지만 각 선택지는 고정적인 크기의 점수를 가지고 있습니다.

• 매우 동의나 매우 비동의 선택지를 선택하면 3점을 얻습니다
• 동의나 비동의 선택지를 선택하면 2점을 얻습니다.
• 약간 동의나 약간 비동의 선택지를 선택하면 1점을 얻습니다.
• 모르겠음 선택지를 선택하면 점수를 얻지 않습니다.

검사 결과는 모든 질문의 성격 유형 점수를 더하여 각 지표에서 더 높은 점수를 받은 성격 유형이 검사자의 성격 유형이라고 판단합니다. 단, 하나의 지표에서 각 성격 유형 점수가 같으면, 두 성격 유형 중 사전 순으로 빠른 성격 유형을 검사자의 성격 유형이라고 판단합니다.

질문마다 판단하는 지표를 담은 1차원 문자열 배열 survey와 검사자가 각 질문마다 선택한 선택지를 담은 1차원 정수 배열 choices가 매개변수로 주어집니다. 이때, 검사자의 성격 유형 검사 결과를 지표 번호 순서대로 return 하도록 solution 함수를 완성해주세요.

 

제한사항

• 1 ≤ survey의 길이 ( = n) ≤ 1,0001 ≤ survey의 길이 ( = n) ≤ 1,000
• survey의 원소는 "RT", "TR", "FC", "CF", "MJ", "JM", "AN", "NA" 중 하나입니다.
• survey[i]의 첫 번째 캐릭터는 i+1번 질문의 비동의 관련 선택지를 선택하면 받는 성격 유형을 의미합니다.
• survey[i]의 두 번째 캐릭터는 i+1번 질문의 동의 관련 선택지를 선택하면 받는 성격 유형을 의미합니다.

 

• choices의 길이 = survey의 길이
• choices[i]는 검사자가 선택한 i+1번째 질문의 선택지를 의미합니다.
• 1 ≤ choices의 원소 ≤ 7

---

Solution

import "strings"

func solution(survey []string, choices []int) string {
	return getPersonalityByPoint(getPointChart(survey, choices))
}

func getPointChart(survey []string, choices []int) map[string]int {

	var result = map[string]int{}

	for i, s := range survey {

		f := s[:1]
		b := s[1:]

		switch {
		case choices[i] == 1:
			result[f] += 3
		case choices[i] == 2:
			result[f] += 2
		case choices[i] == 3:
			result[f] += 1
		case choices[i] == 4:
			continue
		case choices[i] == 5:
			result[b] += 1
		case choices[i] == 6:
			result[b] += 2
		case choices[i] == 7:
			result[b] += 3
		}
	}

	return result
}

func getPersonalityByPoint(points map[string]int) string {
	var result strings.Builder

	for _, t := range []string{"RT", "CF", "JM", "AN"} {
		f := t[:1]
		b := t[1:]

		var pointF int
		var pointB int

		if p, ok := points[f]; ok {
			pointF = p
		}

		if p, ok := points[b]; ok {
			pointB = p
		}

		if pointF > pointB {
			result.WriteString(f)
			continue
		}

		if pointB > pointF {
			result.WriteString(b)
			continue
		}

		if f > b {
			result.WriteString(b)
			continue
		}

		if b > f {
			result.WriteString(f)
			continue
		}
	}

	return result.String()
}