[Python] 백준 3181번(줄임말 만들기) 문제 풀이

 

can_pass = ['i', 'pa', 'te', 'ni', 'niti', 'a', 'ali', 'nego', 'no', 'ili']
a = input().split(' ')
if a[0] not in can_pass:
    a = [item for item in a if item not in can_pass]
else:
    a = sum([[a[0]], [item for item in a if item not in can_pass]], [])
for i in a:
    print(i[0].upper(), end='')

 

 뛸 수 있는 단어들을 리스트로 만들어 놓고, 문자열을 입력받는다. 그 뒤에 입력받은 문자열의 첫 번째 요소가 뛸 수 있는 단어가 아니라면 입력받은 문자열에 있는 모든 단어들 중 뛸 수 있는 단어를 지우고, 입력받은 문자열의 첫 번째 요소가 뛸 수 있는 단어라면 문자열에 있는 모든 단어들 중 뛸 수 있는 단어를 지운 뒤 입력받은 문자열의 첫 번째 요소를 더하여 이차원 배열로 만든 것을 다시 일차원 배열로 만든 뒤에 출력한다.

 

https://developer-next-to-you.tistory.com/12

[Python] 2차원 리스트를 1차원 리스트로