[백준/python] 10798번 세로 읽기:

 

arr=[input() for i in range(5)]

for j in range(15):
    for i in range(5):
            if j<len(arr[i]):
                    print(arr[i][j], end='')

 

리스트 컴프리헨션을 이용해 arr를 한줄에 간단하게 입력받을 수 있음.

arr=['입력1', '입력2', '입력3', '입력4', '입력5'] 로 됨.

 

'''
import numpy as np

arr=[0]*5
for i in range(5):
    arr[i]=list((input()))

print(arr)

arr=arr.T #전치행렬로 바꿔주기. 근데 이건 numpy array만 가능하네..

for i in arr:
    print(i,end='')
'''

 

처음에는 전치행렬로 만들고 2중 for-loop으로 순서대로 쭉 print하려고 했었다.

굳이.. 그럴 필요가 없었다.