BOJ 1614 - 영식이의 손가락 (Python)

S3, 구현

문제 설명

영식이는 숫자를 셀 때, 왼손을 이용한다. 엄지손가락부터 시작해서 새끼손가락까지 차례대로 하나씩 센다. 그다음에 새끼손가락까지 센 다음에는 반대로 엄지손가락으로 다시 역방향으로 센다. 영식이는 자기가 원하는 숫자가 나올 때 까지 계속해서 이 방법으로 센다. 영식이는 절대 손가락을 건너뛰지 않는다. 예를 들어 숫자 10을 셀 때는, 엄지 → 검지 → 중지 → 약지 → 새끼 → 약지 → 중지 → 검지 → 엄지 → 검지 이렇게 센다.

영식이가 손가락을 하나 다쳤다. 영식이는 오른손으로는 셀 수 없기 때문에, 왼손으로 세야 한다. 다친 손가락을 이용해서 셀 수 있는 횟수가 제한되어 있다.

영식이가 셀 수 있는 최대 숫자를 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 영식이가 다친 손가락이 주어진다. 엄지부터 차례대로 1, 2, 3, 4, 5로 번호가 매겨져 있다. 둘째 줄에는 영식이가 다친 손가락으로 몇 번 셀 수 있는지 주어진다. 이 수는 1,000,000,000보다 작거나 같은 자연수 또는 0이다.

출력

첫째 줄에 영식이가 셀 수 있는 수의 최댓값을 출력한다. 만약 시작도 할 수 없으면 0을 출력한다.

입력 예시

2
3

출력 예시

15

풀이

• 단순적으로 구현과정을 통해 해결하고자 했으나
  시간초과가 발생했다.

(시간초과 발생 코드)

from sys import stdin

finger = int(stdin.readline().rstrip())
num = int(stdin.readline().rstrip())

stack = []
flag = False
n = 1
while 1:
    stack.append(n)

    if not flag:
        n += 1
    else:
        n -= 1
    if n == 5:
        flag = True
    elif n == 1:
        flag = False


    if stack.count(finger) == num + 1:
        break

print(len(stack) - 1)

• 다친 손가락의 번호가 1,5 일때와 2,3,4 일때 계산 과정이 다르다.

• 다친 손가락이 1 or 5이면
  1,2,3,4,5,4,3,2의 8개의 반복에서 한 번만 나온다.

• 2, 3, 4이면
  2번씩 나온다.

finger = int(stdin.readline().rstrip())
num = int(stdin.readline().rstrip())
res = 0

if num == 0:
    res = finger - 1
    print(res)
    exit()

if finger == 1:
    res += 8 * num
elif finger == 5:
    res += 4 + 8 * num
else:
    res += 4 * num + 1
    if num % 2 == 0:
        res += finger - 2
    else:
        res += 4 - finger

print(res)