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

프로그래머스

 

문제 푸는 데 있어 도움이 되도록 나의 풀이와 개선된 풀이를 올립니다.
또한 풀이 후 다른 사람의 풀이를 보고 참고할만한 풀이도 올립니다.
- 문제에 따라 나의 풀이만 있을 수 있습니다.
- 해당 풀이들은 풀이 중 하나일 뿐 최선의 풀이는 아닐 수 있습니다.

 

---

 

문제 설명

 

성격유형으로

[R,T],[C,F],[M,J],[N,A]

총 4묶음의 8가지 성격유형이 있습니다.

 

묶음마다의 유형별 점수가 큰 쪽이

적합한 성격 유형이 됩니다.

만약 유형별 점수가 같다면

사전적으로 먼저인 유형이 적합한 성격 유형입니다.

 

만약 R: 3, T: 5 라면

T가 적합한 성격 유형이되며

 

총 4묶음의 적절한 유형을 문자열로 반환하면 됩니다.

 

---

 

나의 풀이

import Foundation

func solution(_ survey:[String], _ choices:[Int]) -> String {
    var personality = ["R":0,"T":0,"C":0,"F":0,"M":0,"J":0,"N":0,"A":0]
    var result = ["R","C","J","A"]
    for i in 0..<survey.count {
        let exam = Array(survey[i])
        switch choices[i] {
        case 1: personality[String(exam[0])]! += 3
        case 2: personality[String(exam[0])]! += 2
        case 3: personality[String(exam[0])]! += 1
        case 4: continue
        case 5: personality[String(exam[1])]! += 1
        case 6: personality[String(exam[1])]! += 2
        case 7: personality[String(exam[1])]! += 3
        default: break
        }
    }
    if personality["R"]! < personality["T"]! { result[0] = "T" }
    if personality["C"]! < personality["F"]! { result[1] = "F" }
    if personality["J"]! < personality["M"]! { result[2] = "M" }
    if personality["A"]! < personality["N"]! { result[3] = "N" }
    return result.joined()
}

 

모든 성격유형을 딕셔너리로 만들고

미리 성격유형의 기본값을 배열로 만들어뒀습니다.

 

survey와 choices는 길이가 같기에

인덱스를 공유하여

적절한유형에 딕셔너리를 찾아 점수를 합해줍니다.

 

다음 묶음마다의 값을 비교하여 사전적단어가 후 순위인 유형이 더 크다면

미리 만든 성격유형의 기본값을 후 순위 유형으로 바꿔 문자열로 반환합니다.

 

---

 

참고할만한 풀이

import Foundation

func solution(_ survey:[String], _ choices:[Int]) -> String {
    let type = [["R", "T"], ["C", "F"], ["J", "M"], ["A", "N"]]
    var score = ["R": 0, "T": 0, "C": 0, "F": 0, "J": 0, "M": 0, "A": 0, "N": 0]
    var result = ""

    for i in 0..<survey.count {
        let first = String(survey[i].first!)
        let last = String(survey[i].last!)

        switch choices[i] {
        case 1...3:
            score[first] = score[first]! + (4 - choices[i])
        case 5...7:
            score[last] = score[last]! + (choices[i] - 4)
        default:
            break
        }
    }

    type.forEach {
        score[$0[0]]! >= score[$0[1]]! ? result.append($0[0]) : result.append($0[1])
    }
    return result
}