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

Weekly Challenge 4

 

Level 1

 

문제 설명

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

아래 표는 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 해주세요.

 

나의 코드

def solution(table, languages, preference):
    '''
    table: 직업군 언어 점수
    languages: 개발자 사용 언어 배열
    preference: 언어 선호도
    '''
    table_ = [t.split() for t in table]
    
    job = {t[0]:0 for t in table_}
    
    for language, pref in zip(languages, preference):
        for t in table_:
            job_name = t[0]
            if language in t:
                job[job_name] += (6 - t.index(language)) * pref
                
    job = sorted(list(job.items()), key = lambda x:(-x[1], x[0]))
    
    return job[0][0]

• 2중 for문 하나로 해결 할 수 있을 것 같은 데, 맘에 들게 짜지는 못한 것 같다.
• table과 languages, preference 배열들을 for문으로 어떻게 연결해야 한 번에 돌릴 수 있는 지를 바로 생각해내지 못해서 저 꼴이 된 것 같다.

---

다른 사람의 풀이

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]

• 비슷하지만 훨씬 짧은 코드
• 나도 이런 식으로 짜고 싶었다.
• 나는 아직도 반복문을 제대로 못쓰고 있구나 싶다.