백준.디지털 티비.2816.py

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

2816번: 디지털 티비

 

import sys
# 원하는 채널로 가서 cnt로 와주는 함수
def write(cnt1, num_str1,cnt2, num_str2):
    for i in range(cnt1):
        sys.stdout.write(num_str1)
    for i in range(cnt2):
        sys.stdout.write(num_str2)

if __name__ == '__main__':
    n, kbs_dic = int(input()), dict({'KBS1': 0, 'KBS2':0})
    for i in range(n):
        channel = sys.stdin.readline().strip()
        if channel in kbs_dic:
            kbs_dic[channel] = i
    
    write(kbs_dic['KBS1'], '1',kbs_dic['KBS1'], '4')
    kbs_dic['KBS2'] = kbs_dic['KBS2'] +1 if kbs_dic['KBS1'] > kbs_dic['KBS2'] else kbs_dic['KBS2']
    write(kbs_dic['KBS2'], '1',kbs_dic['KBS2']-1, '4')