[Swift] 프로그래머스 위클리 챌린지 4주차 직업군 추천하기

 

Problem

 

코딩테스트 연습 - 4주차

---

Solution

 

해당 문제는 Hashtable을 이용해서 풀어야 하는 문제입니다.

 

1. 이중 딕셔너리를 이용해서 직업군에 따른 언어별 점수를 저장한다.

 

table을 공백을 기준으로 나눠줍니다.

 

0번째는 직업군의 이름이므로 키값으로 설정해줍니다.

 

언어별로 점수를 저장해주는데 1번째는 5,2번째는 4,3번째는 3... 이므로 6 - i로 점수를 저장해줍니다.

 

[직업군:[언어:점수]] <- 이런식으로 2중 딕셔너리를 만들어 저장해줍니다.

 

func setJobLanguageScore(table:[String]) -> [String:[String:Int]] {
    var jobs:[String:[String:Int]] = [:]
    table.forEach {
        let split = $0.split(separator: " ")
        let job = String(split[0])
        for (i,lang) in split.enumerated() {
            if i == 0 {
                jobs[job] = [:]
                continue
            }
            jobs[job]![String(lang)] = 6-i
        }
    }
    return jobs
}

 

2. 직업군별 언어 점수 * 언어선호도의 합을 구해준다.

 

위에서 구한 직업군에 따른 언어별 점수 딕셔너리의 키값에 언어를 넣어줘 점수를 구해줍니다.

 

만약 딕셔너리에 해당 언어가 없다면 0점이라는 의미입니다.

 

이런식으로 모든 언어를 순회해 해당 선호도와 언어점수를 곱해줘 합을 구해줍니다.

 

func setJobLangaugeScoreAndPreferenceSum(languages:[String],preference:[Int],jobs:[String:[String:Int]]) -> [String:Int] {
    var scores:[String:Int] = [:]
    for (i,lang) in languages.enumerated() {
        for key in jobs.keys {
            let jobScore = jobs[key]?[lang] == nil ? 0 : jobs[key]![lang]!
            if scores[key] == nil {
                scores[key] = preference[i] * jobScore
            }else {
                scores[key]! += preference[i] * jobScore
            }
        }
    }
    return scores
}

 

3. 가장 점수가 높은 직업군을 구한다.

 

2에서 구한 총합들 중 가장 높은 점수를 찾아 answer에 넣어줍니다.

 

여러개일 경우 사전 순으로 정렬하여 하나만 구해줘야 하기 때문에 정렬한 뒤 가장 첫번째 값을 구해줍니다.

 

func findMaxJob(scores:[String:Int]) -> String {
    var answer:[String] = []
    let max = scores.values.max()
    for score in scores {
        if score.value == max {
            answer.append(score.key)
        }
    }
    return answer.sorted().first!
}

 

4. 가장 점수가 높은 직업군을 반환한다.

 

func solution(_ table:[String], _ languages:[String], _ preference:[Int]) -> String {
    let jobs:[String:[String:Int]] = setJobLanguageScore(table: table)
    let scores:[String:Int] = setJobLangaugeScoreAndPreferenceSum(languages: languages, preference: preference, jobs: jobs)
    return findMaxJob(scores: scores)
}

---

Source Code