프로그래머스 - 양궁대회

양궁대회

설명

카카오배 양궁대회 운영위원회는 한 선수의 연속 우승보다는 다양한 선수들이 양궁대회에서 우승하기를 원합니다. 따라서, 양궁대회 운영위원회는 결승전 규칙을 전 대회 우승자인 라이언에게 불리하게 다음과 같이 정했습니다.
만약, k(k는 1~10사이의 자연수)점을 어피치가 a발을 맞혔고 라이언이 b발을 맞혔을 경우 더 많은 화살을 k점에 맞힌 선수가 k 점을 가져갑니다. 단, a = b일 경우는 어피치가 k점을 가져갑니다. k점을 여러 발 맞혀도 k점 보다 많은 점수를 가져가는 게 아니고 k점만 가져가는 것을 유의하세요. 또한 a = b = 0 인 경우, 즉, 라이언과 어피치 모두 k점에 단 하나의 화살도 맞히지 못한 경우는 어느 누구도 k점을 가져가지 않습니다.
화살의 개수를 담은 자연수 n, 어피치가 맞힌 과녁 점수의 개수를 10점부터 0점까지 순서대로 담은 정수 배열 info가 매개변수로 주어집니다. 이때, 라이언이 가장 큰 점수 차이로 우승하기 위해 n발의 화살을 어떤 과녁 점수에 맞혀야 하는지를 10점부터 0점까지 순서대로 정수 배열에 담아 return 하도록 solution 함수를 완성해 주세요. 만약, 라이언이 우승할 수 없는 경우(무조건 지거나 비기는 경우)는 [-1]을 return 해주세요.

# 중복조합 완전탐색
from itertools import combinations_with_replacement
from collections import Counter, defaultdict
def solution(n, info):
    answer = {}
    info.reverse()
    cases = list(combinations_with_replacement([0,1,2,3,4,5,6,7,8,9,10], n))
    apeach = dict()
    for i in range(11):
        apeach[i] = info[i]
    maxDiff = 0
    bestCase = -1
    for case in cases:
        lion = Counter(case)
        l_score = 0
        a_score = 0
        for i in range(11):
            if lion[i] == 0 and apeach[i] == 0:
                continue
            if lion[i]>apeach[i]:
                l_score += i
            else:
                a_score += i
        diff = l_score-a_score
        if l_score>a_score and diff>maxDiff:
            # print(lion, l_score)
            bestCase = lion
            maxDiff = diff
    for i in range(11):
        answer[i] = 0
    
    if bestCase != -1:
        for i in bestCase.keys():
            answer[i] = bestCase[i]
        answer = list(answer.values())
        answer.reverse()
    else:
        answer = [-1]
        
    return answer

• 보자마자 중복조합을 사용해서 완전탐색을 해야겠다는 생각이 들었다.
• 0~10에서 n개만큼 중복조합을 구한 뒤, 각 case에 대해 라이언이 어피치보다 점수가 높으며, 점수 차이가 가장 큰 case를 탐색하였다.
  • 몇 번째에 몇 점을 맞추든 최종적으로 각 점수에 몇 발이 꽂혔는지만 보면 되므로 중복조합
  • 가장 고득점이 아닌 점수 차이가 가장 큰 case를 찾아야 한다. <- 이거 놓쳐서 고생함
• 각 점수를 계산하기 위해 Counter를 활용하였다.
• 해당 득점에 꽂힌 화살의 수가 라이언이 많으면 라이언에 점수 추가, 둘 다 0점이면 아무도 추가하지 않고, 둘 중 하나라도 꽂혔는데 어피치가 라이언보다 많거나 같으면 어피치에 추가
• 그 다음 라이언이 어피치보다 점수가 높고, 점수 차가 기존 최대 점수 차보다 크다면 갱신한다.
• 최종적으로 최대 점수 차를 낸 case에 대해 출력한다.