동캄의 코딩도장

프로그래머스 level2 [양궁대회] 파이썬 본문

코테/프로그래머스

프로그래머스 level2 [양궁대회] 파이썬

동 캄 2022. 2. 9. 12:12

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

 

코딩테스트 연습 - 양궁대회

문제 설명 카카오배 양궁대회가 열렸습니다. 라이언은 저번 카카오배 양궁대회 우승자이고 이번 대회에도 결승전까지 올라왔습니다. 결승전 상대는 어피치입니다. 카카오배 양궁대회 운영위원

programmers.co.kr

# 프로그래머스 양궁대회
from sys import setrecursionlimit

setrecursionlimit(10**6)


def Do(info, lst, n, i, diff):
    global m
    global score
    if i > 0:
        # 이기는 경우
        if info[i] == 0:
            need = 1
        else:
            need = info[i]+1
        if need <= n:
            if info[i] != 0:
                Do(info, lst+str(need), n-need, i-1, diff-i*2)
            else:
                Do(info, lst+str(need), n-need, i-1, diff-i)
        # 지는 경우
        Do(info, lst+'0', n, i-1, diff)
    elif i == 0:
        Do(info, lst+str(n), 0, i-1, diff)
    else:
        if diff < 0 and diff < score:
            score = diff
            m.clear()
            l = list(map(int, lst))
            l.reverse()
            m.append(l)
        elif diff < 0 and diff == score:
            l = list(map(int, lst))
            l.reverse()
            m.append(l)


def solution(n, info):
    global m
    global score
    score = 10000
    m = []
    answer = []
    info.reverse()
    diff = 0
    for i in range(11):
        if info[i] != 0:
            diff += i
    Do(info, '', n, 10, diff)
    if m:
        m.sort()
        answer = m[-1]
        answer.reverse()
        return answer
    else:
        return [-1]

Do 함수를 이용하여 재귀적으로 라이언이 화살을 쏘는 동작을 두 번 ( 화살을 쏴서 이기는 경우, 화살을 한 번도 쏘지 않고 지는경우)를 반복하도록 구성하였다.

 

값이 갱신되면 그동안의 table(라이언이 쏜 화살의 테이블)을 갱신한다. 동점인 경우는 table을 추가한다.