프로그래머스_Lv1_성격유형검사하기_이승범

seungbum/week5/프로그래머스_Lv1_성격유형검사하기.kt

@@ -0,0 +1,47 @@
+package koala.programmers
+
+fun solution(survey: Array<String>, choices: IntArray): String {
+    val personalType = mutableMapOf(
+        "R" to 0, "T" to 0,
+        "C" to 0, "F" to 0,
+        "J" to 0, "M" to 0,
+        "A" to 0, "N" to 0
+    )
+
+    for (i in choices.indices) {
+        val choice = choices[i]
+        when {
+            (choice == 4) ->
+                continue
+
+            (choice < 4) ->
+                personalType[survey[i][0].toString()] = personalType[survey[i][0].toString()]!! + (4 - choice)
+
+            else ->
+                personalType[survey[i][1].toString()] = personalType[survey[i][1].toString()]!! + choice - 4
+        }
+    }
+    val answer = buildString {
+        append(if (personalType["R"]!! < personalType["T"]!!) "T" else "R")
+        append(if (personalType["C"]!! < personalType["F"]!!) "F" else "C")
+        append(if (personalType["J"]!! < personalType["M"]!!) "M" else "J")
+        append(if (personalType["A"]!! < personalType["N"]!!) "N" else "A")
+    }
+    return answer
+}
+
+fun main() {
+    // Array<String> for survey choices
+    val surveyChoices: Array<String> = arrayOf("AN", "CF", "MJ", "RT", "NA")
+    val surveyChoices2: Array<String> = arrayOf("TR", "RT", "TR")
+
+
+    // IntArray for results
+    val surveyResults: IntArray = intArrayOf(5, 3, 2, 7, 5)
+    val surveyResults2: IntArray = intArrayOf(7, 1, 3)
+
+
+    val result = solution(surveyChoices, surveyResults)
+
+    print(solution(surveyChoices2,surveyResults2))
+}