[ 2023-06-22 😾 TIL ]

백준 1614번 파이썬

문제

코드

import sys

input = sys.stdin.readline
check = [1, 2, 3, 4, 5, 4, 3, 2]
# 위의 숫자가 반복되며 나타남
# 1, 5는 한번, 그 외 숫자는 2번씩 나타나는 것을 확인 가능

f = int(input())
use = int(input())
result = 0

if f == 1 or f == 5 :
	# 한번 씩만 나오는 숫자는 한번 나오면 규칙이 끝나기에 8을 곱해준다. 
    result = (use) * 8
    use = 0
    # f가 1일 경우는 반복문을 안 돌아도 되지만 5일 경우에는 앞에 숫자들이 있어 반복문을 돌려준다
    # 하지만.. 5일 경우에만 +4 해주는 조건으로 변경 가능 할 것 같다.
    for i in range(8):
      if (check[i] == f) and (use == 0):
        break
      result += 1
else :
 # 규칙에서 2번 씩 나오는 숫자들
    result = (use // 2) * 8
    use = (use % 2) 
    # 1로 선언해줘도 될 듯하다.
    for i in range(8):
      if (check[i] == f) and (use == 0):
        break
      elif (check[i] == f):
        use -= 1
      result += 1

print(result)

꼭 블로그 작성하려 하면 아쉬운 점이 보인다..