Algorithm / 성격 유형 검사하기

알고리즘 코드카타

목록 보기
22/59

1) 문제

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

지표 번호성격 유형
1번 지표라이언형(R), 튜브형(T)
2번 지표콘형(C), 프로도형(F)
3번 지표제이지형(J), 무지형(M)
4번 지표어피치형(A), 네오형(N)

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

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

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

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

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

선택지성격 유형 점수
매우 비동의네오형 3점
비동의네오형 2점
약간 비동의네오형 1점
모르겠음어떤 성격 유형도 점수를 얻지 않습니다
약간 동의어피치형 1점
동의어피치형 2점
매우 동의어피치형 3점

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

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

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

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

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

2) 문제 풀이

func solution(_ survey:[String], _ choices:[Int]) -> String {
    let first: [String] = ["R", "C", "J", "A"]
    let second: [String] = ["T", "F", "M", "N"]
    var firstTypes: [String: Int] = [:]
    var secondTypes: [String: Int] = [:]
    
    for type in first {
        firstTypes[type] = 0
    }
    
    for type in second {
        secondTypes[type] = 0
    }
    
    for (index, type) in survey.enumerated() {
        let firstType = String(type.first!)
        let secondType = String(type.last!)
        let count = choices[index]
        
        if count < 4 {
            var value: Int = 0
            
            if count == 1 {
                value = 3
            } else if count == 2 {
                value = 2
            } else if count == 3 {
                value = 1
            }
            
            if firstTypes.keys.contains(firstType) {
                firstTypes[firstType]! += value
            } else if secondTypes.keys.contains(firstType) {
                secondTypes[firstType]! += value
            }
            
        } else if count > 4 {
            var value: Int = 0
            
            if count == 5 {
                value = 1
            } else if count == 6 {
                value = 2
            } else if count == 7 {
                value = 3
            }
            
            if firstTypes.keys.contains(secondType) {
                firstTypes[secondType]! += value
            } else if secondTypes.keys.contains(secondType) {
                secondTypes[secondType]! += value
            }
        }
    }
    
    var index: Int = 0
    var answer: String = ""
    
    while index < first.count {
        let firstCount = firstTypes[first[index]] ?? 0
        let secondCount = secondTypes[second[index]] ?? 0
        
        if firstCount > secondCount {
            answer += first[index]
        } else if firstCount < secondCount {
            answer += second[index]
        } else {
            answer += first[index]
        }
        
        index += 1
    }
    
    return answer
}

결과

3) 코드 개선

문제점 분석

  1. Dictionary를 2개로 나눌 필요가 없음
    • Dictionary<Character, Int>를 사용하여 1개의 Dictionary로 관리
  2. 값 계산 로직이 중복되고 비효율적
    • count == 1 -> 3 이런 부분은 value = abs(count - 4)처럼 간단히 처리 가능
  3. contains(key) 후 강제 언래핑의 위험성
    • dictionary[key]! += value 형태는 key가 없을 경우 크래시 위험 있음
    • 초기화가 되어 있다고 하더라도, 좀 더 안정적인 dictionary[key, default: 0] += value 방식 사용
  4. 최종 비교에서 코드가 장황함
    • max로직을 사용하여 가독성 향상 필요
func solution(_ survey: [String], _ choices: [Int]) -> String {
    let indicators = ["RT", "CF", "JM", "AN"]
    var score: [Character: Int] = [:]

    for (index, types) in survey.enumerated() {
        let choice = choices[index]
        let diff = abs(choice - 4)

        if choice < 4 {
            // 비동의 → 앞쪽 문자 점수 추가
            let type = types.first!
            score[type, default: 0] += diff
        } else if choice > 4 {
            // 동의 → 뒤쪽 문자 점수 추가
            let type = types.last!
            score[type, default: 0] += diff
        }
    }

    // 결과 조합
    var result = ""
    for pair in indicators {
        let first = pair.first!
        let second = pair.last!

        let firstScore = score[first, default: 0]
        let secondScore = score[second, default: 0]

        result += firstScore >= secondScore ? String(first) : String(second)
    }

    return result
}

결과

profile
이유있는 코드를 쓰자!!

0개의 댓글