[프로그래머스] 위클리 챌린지4주차

링크

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

코딩테스트 연습 - 4주차

 

풀이

 

코드가 너무 지저분한것 같다..

다른 분 들이 작성하신 코드를 보면 내가 작성한 코드가 부끄러워진다...

from collections import defaultdict


def make_language_score_data(table):
    language_scores = defaultdict(dict)
    rank_score = [i for i in range(5, 0, -1)]
    for table_data in table:
        table_list = (''.join(table_data)).split(' ')
        job = table_list.pop(0)
        data = dict(zip(table_list, rank_score))
        language_scores[job] = data

    return language_scores


def calc_score(language_scores, input_datas):
    result_company = defaultdict(set)
    for company, language_data in language_scores.items():
        score_sum = 0
        for l, s in input_datas.items():
            try:
                score_sum += language_data[l] * s
            except KeyError:
                score_sum += 0 * s
        result_company[score_sum].add(company)
    result_company = sorted(result_company.items(), key=lambda x: x[0], reverse=True)
    _, companies = next(iter(result_company))
    return sorted(list(companies)).pop(0)


def solution(table, languages, preference):
    languages_scores = make_language_score_data(table)
    return calc_score(languages_scores, dict(zip(languages, preference)))