[백준] 10798번 파이썬(Python)

문제

아직 글을 모르는 영석이가 벽에 걸린 칠판에 자석이 붙어있는 글자들을 붙이는 장난감을 가지고 놀고 있다. 

이 장난감에 있는 글자들은 영어 대문자 ‘A’부터 ‘Z’, 영어 소문자 ‘a’부터 ‘z’, 숫자 ‘0’부터 ‘9’이다. 영석이는 칠판에 글자들을 수평으로 일렬로 붙여서 단어를 만든다. 다시 그 아래쪽에 글자들을 붙여서 또 다른 단어를 만든다. 이런 식으로 다섯 개의 단어를 만든다. 아래 그림은 영석이가 칠판에 붙여 만든 단어들의 예이다. 

A A B C D D
a f z z 
0 9 1 2 1
a 8 E W g 6
P 5 h 3 k x

 

한 줄의 단어는 글자들을 빈칸 없이 연속으로 나열해서 최대 15개의 글자들로 이루어진다. 또한 만들어진 다섯 개의 단어들의 글자 개수는 서로 다를 수 있다. 

심심해진 영석이는 칠판에 만들어진 다섯 개의 단어를 세로로 읽으려 한다. 세로로 읽을 때, 각 단어의 첫 번째 글자들을 위에서 아래로 세로로 읽는다. 다음에 두 번째 글자들을 세로로 읽는다. 이런 식으로 왼쪽에서 오른쪽으로 한 자리씩 이동 하면서 동일한 자리의 글자들을 세로로 읽어 나간다. 위의 그림 1의 다섯 번째 자리를 보면 두 번째 줄의 다섯 번째 자리의 글자는 없다. 이런 경우처럼 세로로 읽을 때 해당 자리의 글자가 없으면, 읽지 않고 그 다음 글자를 계속 읽는다. 그림 1의 다섯 번째 자리를 세로로 읽으면 D1gk로 읽는다. 

그림 1에서 영석이가 세로로 읽은 순서대로 글자들을 공백 없이 출력하면 다음과 같다:

Aa0aPAf985Bz1EhCz2W3D1gkD6x

칠판에 붙여진 단어들이 주어질 때, 영석이가 세로로 읽은 순서대로 글자들을 출력하는 프로그램을 작성하시오.

입력

총 다섯줄의 입력이 주어진다. 각 줄에는 최소 1개, 최대 15개의 글자들이 빈칸 없이 연속으로 주어진다. 주어지는 글자는 영어 대문자 ‘A’부터 ‘Z’, 영어 소문자 ‘a’부터 ‘z’, 숫자 ‘0’부터 ‘9’ 중 하나이다. 각 줄의 시작과 마지막에 빈칸은 없다.

출력

영석이가 세로로 읽은 순서대로 글자들을 출력한다. 이때, 글자들을 공백 없이 연속해서 출력한다. 

---

'문제 해결 아이디어'

IndexError를 어떻게 잡을까 했는데 문제에서 한 줄의 단어의 최대 길이가 주어져있기 때문에 만약 입력으로 들어온 단어의 길이가 최대 길이보다 짧다면, 리스트에 남은 칸은 공백으로 채우기로 했다.

ex) 단어 'IndexError'가 들어와 리스트에 저장된다면 ['I', 'n', 'd', 'e', 'x', 'E', 'r', 'r', 'o', 'r']인데 리스트의 길이를 15개로 맞추기 위해 리스트에 공백 5칸을 추가해준다.

 

'정답 코드'

word = []
height_word = []

for _ in range(5):
    word.append(list(input()))

for i in range(5):
    length = 15 - len(word[i])

    for j in range(length):
        word[i].append("")

for j in range(15):
    for i in range(5):
        height_word.append(word[i][j])

print(*height_word, sep="")

 

'돌아보며'

구현까지 시간이 너무 오래 걸렸다. 아이디어가 쉽게 떠오르지 않았기 때문.

그래도 코딩테스트 대비하려면 몇 시간이 걸려도 혼자 아이디어를 떠올리고 구현해보는 것이 큰 도움이 되는 것 같다.

이번 문제 또한 난항을 겪었지만 리스트 range 관련 Error를 온전히 이해할 수 있어서 나쁘지만은 않은 시간이었다.

이러한 크고 작은 시행착오들이 나중에 꼭 좋은 결과로 돌아오길 바란다. 화이팅!