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

직업군 추천하기

문제 링크

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

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

예를 들면, 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" 중 하나입니다.

나의 풀이

def solution(table, languages, preference):
    lang_prefer = {}
    for lang, prefer in zip(languages, preference):
        lang_prefer[lang] = prefer

    scores = []
    for element in table:
        table2 = element.split()
        sum_score = 0
        for lang in languages:
            if lang in table2:
                score = 6-table2.index(lang)
                sum_score += score * lang_prefer[lang]
        scores.append(sum_score)

    answer = []
    for index, score in enumerate(scores):
        if score == max(scores):
            answer.append(table[index].split()[0])
    answer.sort()
    return answer[0]

• 작성한 코드를 크게 세 문단으로 나눠서 보겠다.
  첫 번째 문단의 경우, languages와 preference의 요소들을 zip() 함수를 이용하여 lang_prefer 변수에 딕셔너리 형태로 저장하였다. 그동안 알고리즘 문제를 풀면서 해보지 않은 방식 이었다.
  두 번째 문단은 table의 인덱스 별 요소들을 split() 함수로 나누고 문제의 규칙에 따라 점수를 계산하여 scores 변수에 할당하였다.
  세 번째 문단은 계산이 완료된 scores의 요소들에 해당하는 언어를 사전순으로 배열하고 배열된 값들의 인덱스 [0] 값을 답으로 return 하는 코드이다. 내 답에서 enumerate도 처음 사용해 본 것 같은데 좀 더 익숙해지도록 연습해야 겠다.

다른 사람의 풀이

def solution(table, languages, preference):
    score = {}
    for t in table:
        for lang, pref in zip(languages, preference):
            if lang in t.split():
                score[t.split()[0]] = score.get(t.split()[0], 0) +  (6 - t.split().index(lang)) * pref
    return sorted(score.items(), key = lambda item: [-item[1], item[0]])[0][0]

• 딕셔너리 관련 함수인 get() 함수와 items()를 사용하였다. get()함수는 value 값을 반환하는 함수이며, a.get('key')와 같은 형태로 사용된다. 딕셔너리 자료형 안에 원하는 key값을 없을때에는 a.get('key', 디폴트값) 과 같이 설정하면 디폴트 값이 반환된다. items() 함수는 a.item()와 같이 사용하면 key, value 값을 [(key1, value1), (key2, value2)...] 튜플로 묶어 객체형태로 반환한다.
  이 답안은 내가 아직 숙련되게 사용하지 못하는 lambda, 딕셔너리, zip 함수 등이 동시에 사용되어 있기때문에 좀 더 복습을 하여 익숙해져야한다.