백준 | 디지털 티비

백준 디지털 티비

KBS1이 첫 번째, KBS2가 두 번째인 입력은 주어지지 않으며, 방법의 길이는 500보다 작으면 된다.

그러므로 1번과 4번 버튼만 사용해서 문제를 해결할 수 있다.
(1) 1번 버튼을 통해 KBS1의 인덱스로 화살표를 이동시키고, 4번 버튼을 통해 KBS1을 첫 번째 위치로 이동시킨다.
(2) 1번 버튼을 통해 KBS2의 인덱스로 화살표를 이동시키고, 4번 버튼을 통해 KBS2을 두 번째 위치로 이동시킨다.
(3) 단, 초기에 KBS1이 KBS2보다 뒤에 위치한 경우, KBS1을 첫 번째 위치로 이동시키며 KBS2의 채널이 1만큼 증가하게 되므로, if 문을 통해 이 부분을 처리해줘야 한다.

이때.. 주어지는 채널이 최대 100개라 했으므로, 최악의 경우에도 방법의 길이가 400을 넘지 않는다!

import sys

n = int(sys.stdin.readline())
channels = [sys.stdin.readline().strip() for _ in range(n)]

kbs1_idx = channels.index('KBS1')
kbs2_idx = channels.index('KBS2')

# 초기 채널에서 KBS1의 인덱스 값이 KBS2보다 크다면?
# KBS1을 첫 번째 채널로 이동시키면서 KBS2의 인덱스 값이 1만큼 증가한다.
if kbs1_idx > kbs2_idx:
    kbs2_idx += 1

# KBS1을 첫 번째 채널로 이동시킨다.
commands = '1' * kbs1_idx + '4' * kbs1_idx
# KBS2를 두 번째 채널로 이동시켜야 하므로, 4번 버튼은 인덱스 값보다 1만큼 적게 사용해야 한다.
commands += '1' * kbs2_idx + '4' * (kbs2_idx - 1)

print(commands)