알고리즘 문제(SOL)

[백준/10789/파이썬] 세로읽기

https://www.acmicpc.net/problem/10798

 

10798번: 세로읽기

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

www.acmicpc.net

Problem

  • 아직 글을 모르는 영석이가 벽에 걸린 칠판에 자석이 붙어있는 글자들을 붙이는 장난감을 가지고 놀고 있다.
  • 심심해진 영석이는 칠판에 만들어진 다섯 개의 단어를 세로로 읽으려 한다. 칠판에 붙여진 단어들이 주어질 때, 영석이가 세로로 읽은 순서대로 글자들을 출력하는 프로그램을 작성하시오.

조건

  •  세로로 읽을 때, 각 단어의 첫 번째 글자들을 위에서 아래로 세로로 읽는다. 다음에 두 번째 글자들을 세로로 읽는다. 이런 식으로 왼쪽에서 오른쪽으로 한 자리씩 이동 하면서 동일한 자리의 글자들을 세로로 읽어 나간다. 위의 그림 1의 다섯 번째 자리를 보면 두 번째 줄의 다섯 번째 자리의 글자는 없다. 이런 경우처럼 세로로 읽을 때 해당 자리의 글자가 없으면, 읽지 않고 그 다음 글자를 계속 읽는다.
  • 총 다섯줄의 입력이 주어진다. 각 줄에는 최소 1개, 최대 15개의 글자들이 빈칸 없이 연속으로 주어진다. 주어지는 글자는 영어 대문자 ‘A’부터 ‘Z’, 영어 소문자 ‘a’부터 ‘z’, 숫자 ‘0’부터 ‘9’ 중 하나이다. 각 줄의 시작과 마지막에 빈칸은 없다.

SOL

이 문제도 그냥, 반복문을 column을 기준으로 돌면 되는거 아닌가?라고 생각하다가는, 생각의 늪에 빠져서 나오지 못할 확률이 좀 큰 문제이다.

column을 기준으로 돌기 위해서는 일단, 리스트의 row의 길이가 일정해야한다. 왜냐하면, 길이가 다 다르기 때문에, 2중반복으로, 돌다보면, out of Index가 필수적으로 발생하게 된다. 

 

물론, Out of index가 발생하면, Error 회피를하는 코드를 짜면되지만, 에러를 회피한다는건 아주 위험한 행동이기 때문에, 다른 방법으로 접근을 해야한다.

 

접근방법

  1. 새로운 리스트를 하나 생성한다.
    • 이때, Lst는 2차원 배열으로, 최대 row의 길이가 15니까, 15개를 생성해준다.
  2. 이중반복을 돌면서, lst[0] = 첫번째 글자, lst[1]=두번째 글자 ...이런식으로 문자열을 저장한다.

예를들어, board=[[A,B,C,D,E] ,[1,2,3,4]]라고 한다면, Lst=[[A,1],[B,2],[C,3],[D,4],[E]]이런식으로 mapping을 시킬거다.

 

코드

"string +="연산은 연산 할때마다, O(N)이 걸리므로, lst에 append해주는 방법으로 문자열 처리를 해줬다.

(사실 큰 차이는 없을것 같긴하다)

import sys
input =sys.stdin.readline

words=[list(map(str,input().rstrip())) for _ in range(5)]
lst=[[] for _ in range(15)]

for i in range(5):
    for j in range(len(words[i])):
        lst[j].append(words[i][j])

new_lst=[]
for c in lst:
    new_lst.append(''.join(c))
#print(new_lst)
#print(len(''.join(new_lst)))
print(''.join(new_lst))