[프로그래머스] 순위 검색

문제

지원서 작성

• 코딩테스트 참여 개발언어 항목에 cpp, java, python 중 하나를 선택해야 합니다.
• 지원 직군 항목에 backend, frontend 중 하나를 선택해야 합니다.
• 지원 경력구분 항목에 junior, senior 중 하나를 선택해야 합니다.
• 선호하는 소울푸드로 chicken과 pizza 중 하나를 선택해야 합니다.

지원자가 지원서에 입력한 4가지의 정보와 획득한 코딩테스트 점수를 하나의 문자열로 구성한 값의 배열 info, 개발팀이 궁금해하는 문의조건이 문자열 형태로 담긴 배열 query가 매개변수로 주어질 때, 각 문의조건에 해당하는 사람들의 숫자를 순서대로 배열에 담아 return 하도록 solution 함수를 완성해 주세요.

제한사항

• info 배열의 크기는 1 이상 50,000 이하입니다.
• info 배열 각 원소의 값은 지원자가 지원서에 입력한 4가지 값과 코딩테스트 점수를 합친 "개발언어 직군 경력 소울푸드 점수" 형식입니다.
  • 개발언어는 cpp, java, python 중 하나입니다.
  • 직군은 backend, frontend 중 하나입니다.
  • 경력은 junior, senior 중 하나입니다.
  • 소울푸드는 chicken, pizza 중 하나입니다.
  • 점수는 코딩테스트 점수를 의미하며, 1 이상 100,000 이하인 자연수입니다.
  • 각 단어는 공백문자(스페이스 바) 하나로 구분되어 있습니다.
• query 배열의 크기는 1 이상 100,000 이하입니다.
• query의 각 문자열은 "[조건] X" 형식입니다.
  • [조건]은 "개발언어 and 직군 and 경력 and 소울푸드" 형식의 문자열입니다.
  • 언어는 cpp, java, python, - 중 하나입니다.
  • 직군은 backend, frontend, - 중 하나입니다.
  • 경력은 junior, senior, - 중 하나입니다.
  • 소울푸드는 chicken, pizza, - 중 하나입니다.
  • '-' 표시는 해당 조건을 고려하지 않겠다는 의미입니다.
  • X는 코딩테스트 점수를 의미하며 조건을 만족하는 사람 중 X점 이상 받은 사람은 모두 몇 명인 지를 의미합니다.
  • 각 단어는 공백문자(스페이스 바) 하나로 구분되어 있습니다.

코드

from itertools import combinations
def solution(info, query):
    dic = {'cpp':1<<8,'java':1<<7,'python':1<<6,
           'backend':1<<5,'frontend':1<<4,
           'junior':1<<3,'senior':1<<2,
           'chicken':1<<1,'pizza':1}
    maps = {}
    for inf in [x.split() for x in info]:
        for i in range(5):
            for comb in combinations(inf[:-1], i):
                tmp = 0
                for c in comb:
                    tmp += dic[c]
                if tmp in maps:
                    maps[tmp].append(int(inf[4]))
                else:
                    maps[tmp] = [int(inf[4])]
    # 각 점수 오름차순
    for item in maps.values():
        item.sort()
    
    answer = []
    for q in [(x.replace('and ', '').split()) for x in query]:
        search = 0
        for s in q[:-1]:
            if s != '-': search += dic[s]
        answer.append(cal(maps[search] if search in maps else [0], int(q[4])))
        
    return answer

def cal(scores, target): # 점수들, 기준점
    left, right = 0, len(scores)
    while left < right:
        mid = (left + right) // 2
        if scores[mid] >= target:
            right = mid
        else:
            left = mid+1
    return len(scores) - left