본문 바로가기
알고리즘/백준온라인저지 (Python)

[백준온라인저지] 2816 - 디지털 티비

by 미래문 2021. 8. 24.
반응형


 

 

https://www.acmicpc.net/problem/2816

 

2816번: 디지털 티비

2012년 12월 31일 새벽 4시부터 지상파 아날로그 TV방송이 종료되었다. TV를 자주보는 할머니를 위해서, 상근이네 집도 디지털 수신기를 구입했다. 원래 상근이네 집에는 KBS1과 KBS2만 나왔다. 할머니

www.acmicpc.net

 

import sys

cs = []

result = ''

n = int(sys.stdin.readline())

for i in range(n):
    c = input()
    cs.append(c)

kbs1_index = cs.index('KBS1')

for i in range(kbs1_index):
    result += '1'

for i in range(kbs1_index):
    result += '4'

for i in range(kbs1_index, 0, -1):
    tmp = cs[i-1]
    cs[i-1] = cs[i]
    cs[i] = tmp

kbs2_index = cs.index('KBS2')

for i in range(kbs2_index):
    result += '1'

for i in range(kbs2_index-1):
    result += '4'

print(result)
반응형

댓글