티스토리 뷰
반응형
문제
문자열을 입력으로 주면 문자열의 첫 글자와 마지막 글자를 출력하는 프로그램을 작성하시오.
입력
입력의 첫 줄에는 테스트 케이스의 개수 T(1 ≤ T ≤ 10)가 주어진다. 각 테스트 케이스는 한 줄에 하나의 문자열이 주어진다. 문자열은 알파벳 A~Z 대문자로 이루어지며 알파벳 사이에 공백은 없으며 문자열의 길이는 1000보다 작다.
출력
각 테스트 케이스에 대해서 주어진 문자열의 첫 글자와 마지막 글자를 연속하여 출력한다.
예제 입력 1 복사
3
ACDKJFOWIEGHE
O
AB
예제 출력 1 복사
AE
OO
AB
풀이
첫 줄에는 줄 입력이 주어지므로 줄의 숫자를 입력받고, 그만큼 range로 해서 for를 돌리면 단어들이 들어간 리스트가 만들어진다.
그 다음에 해당 리스트에서 첫줄과 마지막 출을 출력해주기만 하면 된다.
t=int(input())
a = [list(input()) for i in range(t)]
for j in a:
print(j[0],end='')
print(j[-1])
중요한 개념은 두 개이다.
python에서는 리스트를 만들 때 for를 포함해서 리스트를 만들 수 있다는 것.
위의 예시처럼 [(원하는 함수) for (변수) in (범위)] 로 하면 된다. 그러면 해당 함수가 for 문만큼 돌아간 리스트가 만들어진다.
두 번째 개념은 인덱싱에서 [-1]은 끝 번호와 같다는 것.
굳이 끝 번호를 구할 필요가 없어지므로 참고하자.
반응형