[BOJ] 17281. ⚾ (python)

✏️ 문제

 

⚾는 9명으로 이루어진 두 팀이 공격과 수비를 번갈아 하는 게임이다. 하나의 이닝은 공격과 수비로 이루어져 있고, 총 N이닝 동안 게임을 진행해야 한다. 한 이닝에 3아웃이 발생하면 이닝이 종료되고, 두 팀이 공격과 수비를 서로 바꾼다.
두 팀은 경기가 시작하기 전까지 타순(타자가 타석에 서는 순서)을 정해야 하고, 경기 중에는 타순을 변경할 수 없다. 9번 타자까지 공을 쳤는데 3아웃이 발생하지 않은 상태면 이닝은 끝나지 않고, 1번 타자가 다시 타석에 선다. 타순은 이닝이 변경되어도 순서를 유지해야 한다. 예를 들어, 2이닝에 6번 타자가 마지막 타자였다면, 3이닝은 7번 타자부터 타석에 선다.
공격은 투수가 던진 공을 타석에 있는 타자가 치는 것이다. 공격 팀의 선수가 1루, 2루, 3루를 거쳐서 홈에 도착하면 1점을 득점한다. 타자가 홈에 도착하지 못하고 1루, 2루, 3루 중 하나에 머물러있을 수 있다. 루에 있는 선수를 주자라고 한다. 이닝이 시작될 때는 주자는 없다.
타자가 공을 쳐서 얻을 수 있는 결과는 안타, 2루타, 3루타, 홈런, 아웃 중 하나이다. 각각이 발생했을 때, 벌어지는 일은 다음과 같다.
안타: 타자와 모든 주자가 한 루씩 진루한다.
2루타: 타자와 모든 주자가 두 루씩 진루한다.
3루타: 타자와 모든 주자가 세 루씩 진루한다.
홈런: 타자와 모든 주자가 홈까지 진루한다.
아웃: 모든 주자는 진루하지 못하고, 공격 팀에 아웃이 하나 증가한다.
한 야구팀의 감독 아인타는 타순을 정하려고 한다. 아인타 팀의 선수는 총 9명이 있고, 1번부터 9번까지 번호가 매겨져 있다. 아인타는 자신이 가장 좋아하는 선수인 1번 선수를 4번 타자로 미리 결정했다. 이제 다른 선수의 타순을 모두 결정해야 한다. 아인타는 각 선수가 각 이닝에서 어떤 결과를 얻는지 미리 알고 있다. 가장 많은 득점을 하는 타순을 찾고, 그 때의 득점을 구해보자.

 

🤖 알고리즘

 

#구현  #브루트포스

 

 

🤯 풀이 방법

가능한 모든 타순 배열 구해서 냅다 구현 함수 돌림
python 말고 pypy로만 되는 문제라는데 다른 정답 코드들 거의 같은데도 자꾸 시간초과 떠서 봤더니
홈런인 경우 현재 진루 상황 나타내는 배열에서 True(1)값을 count 메소드 이용해서 그 개수만큼 점수 더하려고 했더니 시간초과 뜨고
if문으로 현재 진루 상황 하나하나 돌려서 점수 입력했더니 시간초과 안남

 

👾 구현 코드 (파이썬)

import sys
input=sys.stdin.readline
from itertools import permutations


def play(ord):  # ord: 타순
    ord.insert(3, 0)
    score = 0
    inn = 0  # 현재 이닝
    cur = 0  # 현재 타순
    board = [0, 0, 0]  # 1루, 2루, 3루
    out = 0

    while inn < N:  # 최대 이닝 전까지 진행
        # out 3개 이하인 경우 경기 진행
        # 경기 진행 과정
        res = R[inn][ord[cur]]

        if res == 0:   # 아웃
            out += 1

        elif res == 1:  # 안타
            if board[2] == 1:  # 3루에 주자 있는 경우
                score += 1
                board[2] = 0
            if board[1] == 1:  # 2루에 주자 있는 경우
                board[2] = 1
                board[1] = 0
            if board[0] == 1:  # 1루에 주자 있는 경우
                board[1] = 1
                board[0] = 0
            board[0] = 1  # 타자 진루

        elif res == 2:  # 2루타 (2, 3루 득점 가능)
            if board[2] == 1:  # 3루에 주자 있는 경우
                score += 1
                board[2] = 0
            if board[1] == 1:  # 2루에 주자 있는 경우
                score += 1
                board[1] = 0
            if board[0] == 1:  # 1루에 주자 있는 경우
                board[2] = 1
                board[0] = 0
            board[1] = 1  # 타자 진루 (2루)

        elif res == 3:  # 3루타 (1, 2, 3루 득점 가능)
            if board[2] == 1:  # 3루에 주자 있는 경우
                score += 1
                board[2] = 0
            if board[1] == 1:  # 2루에 주자 있는 경우
                score += 1
                board[1] = 0
            if board[0] == 1:  # 1루에 주자 있는 경우
                score += 1
                board[0] = 0
            board[2] = 1  # 타자 진루 (3루)

        elif res == 4:  # 홈런
            if board[2] == 1:
                score += 1
                board[2] = 0
            if board[1] == 1:
                score += 1
                board[1] = 0
            if board[0] == 1:
                score += 1
                board[0] = 0
            score += 1

            ''' 이렇게 하면 시간초과
            score += board.count(1) + 1  # 타자 포함 모두 득점
            board = [0, 0, 0]
            '''



        cur += 1
        if cur == 9:
            cur = 0


        # out이 3개 되는 순간 다음 이닝
        if out == 3:
            out = 0
            board = [0, 0, 0]
            inn += 1

    return score




N = int(input())  # 이닝 수 (2 <= N <= 50)
L = permutations([1, 2, 3, 4, 5, 6, 7, 8], 8)

R = [list(map(int, input().split())) for _ in range(N)]

ans = 0
for l in L:
    ans = max(ans, play(list(l)))

print(ans)

 

 

17281번: ⚾