구현)2290.LCD Test

문제 출처 : https://www.acmicpc.net/problem/2290

2290번: LCD Test

솔루션 : 

1. 각 패널별, 좌우가 나눠져 있으면 좌우로 나눠 총 7개의 경우의 수로 나눠 풀었다.

2. 각 숫자가 각 패널의 도형이 필요하면 그것을 취한 뒤 최종적으로 숫자 형태로 완성된다.

 

import sys

input = sys.stdin.readline

s,tgs = map(str,input().split())
s=int(s)
hrz = s+2
ver = 2*s+3

h='-'
v='|'

def top(num):
    # num[0][1:-1] = h
    for i in range(1,s+1):
        num[0][i] = h
    return num

def tbml(num):
    for i in range(1,s+1):
        num[i][0] = v
    return num

def tbmr(num):
    for i in range(1,s+1):
        num[i][s+1] = v
    return num

def mid(num):
    # num[s+1][1:-1] = h
    for i in range(1, s + 1):
        num[s + 1][i] = h
    return num

def tbbl(num):
    for i in range(s+2,ver-1):
        num[i][0] = v
    return num

def tbbr(num):
    for i in range(s+2,ver-1):
        num[i][s+1] = v
    return num

def bot(num):
    for i in range(1, s + 1):
        num[ver - 1][i] = h
    # num[ver-1][1:-1] = h
    return num

nums=[]
for tg in tgs:
    tg=int(tg)
    num = [[' '] * (hrz) for _ in range(ver)]
    if tg in [2,3,5,6,7,8,9,0]:
        num = top(num)
    if tg in [4,5,6,8,9,0]:
        num = tbml(num)
    if tg in [1,2,3,4,7,8,9,0]:
        num = tbmr(num)
    if tg in [2,3,4,5,6,8,9]:
        num = mid(num)
    if tg in [2,6,8,0]:
        num = tbbl(num)
    if tg in [1,3,4,5,6,7,8,9,0]:
        num = tbbr(num)
    if tg in [2,3,5,6,8,9,0]:
        num = bot(num)
    nums.append(num)

ans = [[] for _ in range(2*s + 3)]

for i in range(0, 2*s + 3):
    for idx in range(len(nums)):
        ans[i] += nums[idx][i]
        ans[i].append(' ')

for tg in ans:
    print(''.join(tg))