[Baekjoon] 10798 - 세로읽기 (Python)

문제

[BaekJoon] 10798 세로읽기

풀이

in_words = []
str = ""

for _ in range(5):
    s = input()
    in_words.append(s)

for i in range(len(in_words)):
    for word in in_words:
        str += word[i]

print(str)

위 코드는 모든 줄의 단어 길이가 같을 때만 작동함

이 문제는 없는 index는 건너뛰고 다음 줄을 읽어야한다.

in_words = []

for _ in range(5):
    s = input()
    in_words.append(s)

for i in range(max(len(w) for w in in_words)):
    for j in range(len(in_words)):
        if i < len(in_words[j]):
            print(in_words[j][i], end='')