백준 9086번

안녕하세요 :)
오늘은 백준 9086번을 가지고 왔습니다.

이 문제의 핵심은 여러 문자열을 받고 문자열의 길이를 구하여 첫번째 문자와 마지막 문자를 형식에 맞게 출력하는 것입니다.

저는 2차원 배열을 사용해서 문제를 해결했습니다. 문자열을 입력받을 때마다 안쪽 리스트에 넣고 전체 문자열을 바깥 리스트에 저장합니다.
(바깥 리스트는 l이고 안쪽 리스트는 s입니다)

l=[]
for x in range(0, n):
    s = list(sys.stdin.readline())
    l.append(s)

그 다음은 첫번째 문자를 출력하고 안쪽 리스트를 통해 문자열의 길이를 구하고 마지막 문자를 출력합니다. 변수 b는 그 바로 전 문자를 저장하는 변수입니다.

b = ""
for x in l:
    print(x[0], end="")
    for y in x:
        if(y == '\n'):
            print(b, end="\n")
            break
        b = y

다음은 9086번 정답 코드입니다.

import sys

n = int(sys.stdin.readline())

l=[]
for x in range(0, n):
    s = list(sys.stdin.readline())
    l.append(s)

b = ""
for x in l:
    print(x[0], end="")
    for y in x:
        if(y == '\n'):
            print(b, end="\n")
            break
        b = y