자기계발
백준
9086 · 문자열

백준 9086번 문자열

문제

첫 줄에 테스트 케이스 수 T(1 이상 10 이하)가 주어지고, 이어서 알파벳 대문자로만 이루어진 길이 1000 미만의 문자열이 한 줄에 하나씩 주어진다. 각 문자열의 첫 글자와 마지막 글자를 이어서 출력한다. 예를 들어 ACDKJFOWIEGHE는 AE, 한 글자인 O는 OO다.

풀이

문자열의 첫 글자는 word[0], 마지막 글자는 word[-1]로 가져온다. 이 둘을 더하는 함수를 만들고, 케이스 수만큼 반복문을 돌며 입력을 넘겨 출력한다. 한 글자짜리 문자열은 [0]과 [-1]이 같은 글자를 가리키므로 별도 처리가 필요 없다.

def print_(word):
    """
    단어의 첫 글자와 마지막 글자만 뽑는 함수
    """
    start = word[0]
    end = word[-1]
    
    return start + end
 
cnt = int(input())
 
# 처음 주어진 횟수만큼 루프문 생성
for _ in range(cnt):
    start_end = print_(input())
    print(start_end)

출처