[백준 10798번] 세로 읽기 파이썬

백준 단계별로 문제를 풀기 시작하다보면, 2d-array에 관한 문제가 나옵니다.

 

단계별로 올라갈수록 맛보기 수준의 문제에서 난이도가 올라가는 것을 체감할 수 있습니다.

 

오늘은 10798번 문제에 대해 어떻게 풀었는지 공유해보겠습니다. (https://www.acmicpc.net/problem/10798)

 

 

다음 문제에서 챌린지는 

• matrix를 가로방향이 아닌 세로방향으로 읽어야함
• 각 행(단어)별로 0~15까지 가변 길이를 갖고 있음

문제는 두번째 포인트인데, 어떻게 예외를 처리할까 고민하게 될 수 있습니다.

답은 생각보다 간단합니다.

 

if __name__ == '__main__':

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

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

 

따로 예외처리를 힘들게 할 필요 없이, 행의 최대 길이 len(words[i])가 현 iteration 횟수보다 낮으면 출력하는 방식으로 진행할 수 있습니다.