[프로그래머스] 직업군 추천하기

https://programmers.co.kr/learn/courses/30/lessons/84325

코딩테스트 연습 - 4주차

문제 설명

개발자가 사용하는 언어와 언어 선호도를 입력하면 그에 맞는 직업군을 추천해주는 알고리즘을 개발하려고 합니다.

아래 표는 5개 직업군 별로 많이 사용하는 5개 언어에 직업군 언어 점수를 부여한 표입니다.

점수	SI	CONTENTS	HARDWARE	PORTAL	GAME
5	JAVA	JAVASCRIPT	C	JAVA	C++
4	JAVASCRIPT	JAVA	C++	JAVASCRIPT	C#
3	SQL	PYTHON	PYTHON	PYTHON	JAVASCRIPT
2	PYTHON	SQL	JAVA	KOTLIN	C
1	C#	C++	JAVASCRIPT	PHP	JAVA

예를 들면, SQL의 SI 직업군 언어 점수는 3점이지만 CONTENTS 직업군 언어 점수는 2점입니다. SQL의 HARDWARE, PORTAL, GAME 직업군 언어 점수는 0점입니다.

직업군 언어 점수를 정리한 문자열 배열 table, 개발자가 사용하는 언어를 담은 문자열 배열 languages, 언어 선호도를 담은 정수 배열 preference가 매개변수로 주어집니다. 개발자가 사용하는 언어의 언어 선호도 x 직업군 언어 점수의 총합이 가장 높은 직업군을 return 하도록 solution 함수를 완성해주세요. 총합이 같은 직업군이 여러 개일 경우, 이름이 사전 순으로 가장 빠른 직업군을 return 해주세요.

---

제한사항

• table의 길이 = 5
  • table의 원소는 "직업군 5점언어 4점언어 3점언어 2점언어 1점언어"형식의 문자열입니다. 직업군, 5점언어, 4언어, 3점언어, 2점언어, 1점언어는 하나의 공백으로 구분되어 있습니다.
  • table은 모든 테스트케이스에서 동일합니다.
• 1 ≤ languages의 길이 ≤ 9
  • languages의 원소는 "JAVA", "JAVASCRIPT", "C", "C++" ,"C#" , "SQL", "PYTHON", "KOTLIN", "PHP" 중 한 개 이상으로 이루어져 있습니다.
  • languages의 원소는 중복되지 않습니다.
• preference의 길이 = languages의 길이
  • 1 ≤ preference의 원소 ≤ 10
• preference의 i번째 원소는 languages의 i번째 원소의 언어 선호도입니다.
• return 할 문자열은 "SI", "CONTENTS", "HARDWARE", "PORTAL", "GAME" 중 하나입니다.

---

입출력 예

 

table	languages	preference	result
["SI JAVA JAVASCRIPT SQL PYTHON C#", "CONTENTS JAVASCRIPT JAVA PYTHON SQL C++", "HARDWARE C C++ PYTHON JAVA JAVASCRIPT", "PORTAL JAVA JAVASCRIPT PYTHON KOTLIN PHP", "GAME C++ C# JAVASCRIPT C JAVA"]	["PYTHON", "C++", "SQL"]	[7, 5, 5]	"HARDWARE"
["SI JAVA JAVASCRIPT SQL PYTHON C#", "CONTENTS JAVASCRIPT JAVA PYTHON SQL C++", "HARDWARE C C++ PYTHON JAVA JAVASCRIPT", "PORTAL JAVA JAVASCRIPT PYTHON KOTLIN PHP", "GAME C++ C# JAVASCRIPT C JAVA"]	["JAVA", "JAVASCRIPT"]	[7, 5]	"PORTAL"

---

입출력 예 설명

입출력 예 #1

각 직업군 별로 점수를 계산해보면 아래와 같습니다.

아래 사진은 개발자 언어 선호도 나타낸 표입니다.

아래 사진은 개발자가 선호하는 언어의 직업군 언어 점수를 나타낸 표입니다.

따라서 점수 총합이 41로 가장 높은 "HARDWARE"를 return 해야 합니다.

입출력 예 #2

각 직업군 별로 점수를 계산해보면 아래와 같습니다.

아래 사진은 개발자 언어 선호도 나타낸 표입니다.

아래 사진은 개발자가 선호하는 언어의 직업군 언어 점수를 나타낸 표입니다.

점수 총합이 55로 가장 높은 직업군은 "SI" 와 "PORTAL"입니다.
따라서 사전 순으로 먼저 오는 "PORTAL"을 return 해야 합니다.

 

문제풀이

1. 결과값을 return하는 변수(answer), 파라미터로 받은 직업군 언어 점수를 정리한 문자열 배열(table)을 직업군별로 나눈 2차원 배열(table), 2차원 배열(table)의 인덱스 위치 변수(pos), 각 직업군별 언어 점수를 모아 놓은 2차원 배열(languages_point), 각 직업군별 점수를 모아 놓은 딕셔너리(occupation_point) 등을 선언합니다.

2. 첫 번째 for문에서는 개발자가 선호하는 언어의 직업군 언어 점수를 구현합니다.

     2-1. 언어의 갯수만큼 파라미터로 받은 languages배열을 반복합니다.

     2-2. 임시배열(tmp)을 생성하고, 2차원 배열(table)을 반복합니다.

     2-3. 위치 변수(pos)를 0으로 초기화하고 2차원 배열의 요소 값을 반복합니다.

     2-4. 개발자가 사용하는 언어를 찾기 전까지 위치 변수(pos)를 1씩 증가 시키고, 찾으면 for문을 빠져나옵니다.

     2-5. 임시배열(tmp)에 2차원 배열(table)의 갯수(table.count)에서 위치(pos)만큼 뺀 값이 각 직업군별 언어 점수 입니다.

     2-6. 각 직업군별 언어 점수를 모아 놓기 위해 선언한 2차원 배열(languages_point)에 임시배열(tmp)를 추가합니다.

3. 두 번째 for문에서는 각 직업군별 점수를 구현합니다.

     3-1. 각 직업군별 언어 점수를 모아 놓은 2차원 배열(languages_point)를 반복합니다.

     3-2. 2차원 배열(languages_point)의 요소 값들을 반복합니다.

     3-3. 선호도 배열(preference)을 반복하여 개발자가 사용하는 언어의 언어 선호도 x 직업군 언어 점수의 총합을 구합니다.

     3-4. 각 직업군별 점수를 모아 놓가 위해 선언한 딕셔너리(occupation_point)에 키는 직업군으로하고, 값은 개발자가 사용하는 언어의 언어 선호도 x 직업군 언어 점수의 총합으로 추가합니다.

4. for문을 이용하여 각 직업군별 점수를 모아 놓은 딕셔너리(occupation_point)에서 가장 높은 값을 구하고, 가장 높은 값의 키 값 즉, 직업군을 결과 반환 값에 저장합니다. 총합 점수가 같을 시 사전 순으로 먼저 오는 값을 반환합니다.

5. 결과를 반환합니다.

func solution(_ table:[String], _ languages:[String], _ preference:[Int]) -> String {
    var answer = ""
    let table = table.map{$0.split(separator: " ")}
    var pos = 0
    var languages_point = [[Int]]()
    var occupation_point = [String: Int]()

    // 각 직업군별 언어 점수
    for i in 0..<languages.count {
      var tmp = [Int]()
      for item in table {
        pos = 0
        for j in 1..<item.count {
          if item[j].elementsEqual(languages[i]) {
          	break
          }
          	pos += 1
          }
        tmp.append(table.count - pos)
      }
      languages_point.append(tmp)
    }

    // 각 직업군별 점수
    for item in languages_point {
      for i in 0..<item.count {
        var sum = 0
        for j in 0..<preference.count {
        	sum += preference[j] * languages_point[j][i]
        }
        occupation_point.updateValue(sum, forKey: String(table[i][0]))
      }
    }

    // 가장 높은 직업군별 점수
    for (key,value) in occupation_point {
      if value == occupation_point.values.max() {
        if answer.isEmpty {
          	answer = key
          } else {
            if answer > key {
              answer = key
            }
        }
      }
    }

    return answer				
}