아직 글을 모르는 영석이가 벽에 걸린 칠판에 자석이 붙어있는 글자들을 붙이는 장난감을 가지고 놀고 있다.
이 장난감에 있는 글자들은 영어 대문자 ‘A’부터 ‘Z’, 영어 소문자 ‘a’부터 ‘z’, 숫자 ‘0’부터 ‘9’이다. 영석이는 칠판에 글자들을 수평으로 일렬로 붙여서 단어를 만든다. 다시 그 아래쪽에 글자들을 붙여서 또 다른 단어를 만든다. 이런 식으로 다섯 개의 단어를 만든다. 아래 그림 1은 영석이가 칠판에 붙여 만든 단어들의 예이다.
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
<그림 1>
한 줄의 단어는 글자들을 빈칸 없이 연속으로 나열해서 최대 15개의 글자들로 이루어진다. 또한 만들어진 다섯 개의 단어들의 글자 개수는 서로 다를 수 있다.
심심해진 영석이는 칠판에 만들어진 다섯 개의 단어를 세로로 읽으려 한다. 세로로 읽을 때, 각 단어의 첫 번째 글자들을 위에서 아래로 세로로 읽는다. 다음에 두 번째 글자들을 세로로 읽는다. 이런 식으로 왼쪽에서 오른쪽으로 한 자리씩 이동 하면서 동일한 자리의 글자들을 세로로 읽어 나간다. 위의 그림 1의 다섯 번째 자리를 보면 두 번째 줄의 다섯 번째 자리의 글자는 없다. 이런 경우처럼 세로로 읽을 때 해당 자리의 글자가 없으면, 읽지 않고 그 다음 글자를 계속 읽는다. 그림 1의 다섯 번째 자리를 세로로 읽으면 D1gk로 읽는다.
그림 1에서 영석이가 세로로 읽은 순서대로 글자들을 공백 없이 출력하면 다음과 같다:
Aa0aPAf985Bz1EhCz2W3D1gkD6x
칠판에 붙여진 단어들이 주어질 때, 영석이가 세로로 읽은 순서대로 글자들을 출력하는 프로그램을 작성하시오.
총 다섯줄의 입력이 주어진다. 각 줄에는 최소 1개, 최대 15개의 글자들이 빈칸 없이 연속으로 주어진다. 주어지는 글자는 영어 대문자 ‘A’부터 ‘Z’, 영어 소문자 ‘a’부터 ‘z’, 숫자 ‘0’부터 ‘9’ 중 하나이다. 각 줄의 시작과 마지막에 빈칸은 없다.
영석이가 세로로 읽은 순서대로 글자들을 출력한다. 이때, 글자들을 공백 없이 연속해서 출력한다.
ABCDE
abcde
01234
FGHIJ
fghij
Aa0FfBb1GgCc2HhDd3IiEe4Jj
입력이 5줄이 주어진다.
words = []
for i in range(5):
a = input()
words.append(a)
위의 코드는 내가 항상 반복문으로 입력받는 코드이다.
이것보다 더 간단한 코드가 있었다.
words = [input() for i in range(5)]
네 줄로 쓰던 코드가 한 줄로 줄었다.
올해 배웠는데 까먹었다
이것은 리스트 컴프리헨션(list comprehension)으로 리스트를 만드는 간결한 방법을 제공한다.
입력을 다 받았으면 출력을 해야 한다.
다섯 개의 단어를 글자 하나씩 세로로 읽어야 한다.
첫 번째로 읽을 단어는 words[0][0]
이고 두 번째는 words[1][0]
이다. 이렇게 위에서 아래로, 왼쪽에서 오른쪽으로 읽어나간다. 문제는 단어의 길이가 같지 않다는 것이다. 단어의 최대 길이는 15이다.
글자를 출력하기 위한 반복문은
안쪽 반복문은 5번 반복하고
바깥쪽 반복문은 숫자가 크지 않으므로 15번 반복한다.
안쪽 반복문의 변수는 편의를 위해 i, 바깥쪽 반복문의 변수는 j로 한다. 출력할 때 print(words[i][j])
로 하기 위해서이다.
j가 단어 길이보다 작으면 인덱스 범위 안에 있으므로 출력할 수 있다.
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='')
솔직히 어려웠다. 푸는데 시간이 꽤 걸렸고 예외 처리를 하려다 아직 공부 안 한 부분이어서 고민하다가 겨우 풀었다. 나중에 이 글을 보다가 더 좋은 풀이가 있다면 다시 와서 고쳐야겠다.