본문 바로가기

코딩 테스트/프로그래머스

[Python] 백준 9086 문자열

 

정답 코드 1

tc = int(input())
for _ in range(tc) :
    str = input()
    print(str[0]+str[-1])

 

그런데 원래 입력 받던 것 처럼

import sys
input = sys.stdin.readline

tc = int(input())
for _ in range(tc) :
    str = input()
    print(str[0]+str[-1])
    

# 출력
# A
# 
# O
#
#
# A
#
#

로 입력 코드를 작성하면 저러한 출력 결과가 나온다.

저렇게 입력을 받으면 문자열 끝에 '\n'이 포함되기 때문

따라서 다음과 같이 코드를 작성해줘야한다.

 

정답 코드 2

import sys
input = lambda : sys.stdin.readline().strip()

tc = int(input())
for _ in range(tc) :
    str = input()
    print(str[0]+str[-1])