[백준] 문자열 (Python - 9086번)

문제

문자열을 입력으로 주면 문자열의 첫 글자와 마지막 글자를 출력하는 프로그램을 작성하시오.

입력

입력의 첫 줄에는 테스트 케이스의 개수 T(1 ≤ T ≤ 10)가 주어진다. 각 테스트 케이스는 한 줄에 하나의 문자열이 주어진다. 문자열은 알파벳 A~Z 대문자로 이루어지며 알파벳 사이에 공백은 없으며 문자열의 길이는 1000보다 작다.

출력

각 테스트 케이스에 대해서 주어진 문자열의 첫 글자와 마지막 글자를 연속하여 출력한다.

예제 입력 1

3
ACDKJFOWIEGHE
O
AB

예제 출력 1

AE
OO
AB

되게 무식하게 푼것 같긴하지만

일단 T만큼 for문으로 반복시키면서 알파벳을 alpha에 담아주고, 그 알파벳의 길이를 length에 선언 후
첫번째 알파벳 변수, 마지막 알파벳 변수를 각각 담아서 print로 +해서 출력하면 풀렸다

최종 코드

T = int(input())
for _ in range(T):
  alpha = input()
  length = len(alpha)
  first = alpha[0]
  last = alpha[length-1]
  print(first+last)