[알고리즘/구현] 17281번 : ⚾ - python

링크 : https://www.acmicpc.net/problem/17281

17281번: ⚾

 

---

• 문제
• 소요시간: 1시간 30분

---

• 설계하기(접근방법) 

1. 입력받기

이닝별 타자의 결과를 입력받는다

 

2. 구현하기

 

타순은 4번째 타자를 1번으로 고정을 하고

permutations로 

 

base를 구현하는 것이 까다로웠다

 

 

3. 출력하기

 

 

---

• 코드(출력)

import sys
from itertools import permutations

input = sys.stdin.readline

def find_score(order, n, result):
    score = 0
    p = 0  # 타자 순서
    for inning in range(n):
        out_count = 0
        base = [0, 0, 0]
        while out_count < 3:
            player_now = order[p] - 1
            record = result[inning][player_now]
            if record == 0:
                out_count += 1
            elif record == 4:
                score += 1 + sum(base)
                base = [0, 0, 0]
            else:
                for i in range(2, -1, -1):
                    if base[i] == 1:
                        if i + record >= 3:
                            score += 1
                        else:
                            base[i + record] = 1
                        base[i] = 0
                base[record - 1] = 1
            p = (p + 1) % 9

    return score

# 이닝
n = int(input())

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

max_score = 0

fixed_order = [1] * 9
for i, perm in enumerate(permutations(range(2, 10), 8), start=1):
    order = list(perm[:3]) + [1] + list(perm[3:])
    max_score = max(max_score, find_score(order, n, result))

print(max_score)

---

• 얻어갈 부분

1. 시간초과가 난 코드이다. 개념만 익혀놓자

 if inning[p] == 0:
                    out += 1
                elif inning[p] == 1:
                    score += b3
                    b1, b2, b3 = 1, b1, b2
                elif inning[p] == 2:
                    score += b2 + b3
                    b1, b2, b3 = 0, 1, b1
                elif inning[p] == 3:
                    score += b1 + b2 + b3
                    b1, b2, b3 = 0, 0, 1
                else:
                    score += b1 + b2 + b3 + 1
                    b1, b2, b3 = 0, 0, 0

이와 같은 형태로 구현하는 것도 직관적이었을 것 같다