[백준/Python 3] 3181번: 줄임말 만들기

문제

 

 

코드

arr = list(input().split())
arr1 = []
for i in arr:
    if i == 'i' or i == 'pa' or i == 'te' or i == 'ni' or i == 'niti' or i == 'a' or i == 'ali' or i == 'nego' or i == 'no' or i == 'ili':
        if i == arr[0]:
            arr1.append(i)
        else:
            continue
    else:
        arr1.append(i)

ans = []
for i in range(len(arr2)):
    ans.append(arr1[i][0].upper())
    
print(''.join(ans))

 

해설

입력되는 글자를 공백에 따라 arr이라는 리스트에 저장한다. 'i', 'pa', 'te', 'ni', 'niti', 'a', 'ali', 'nego', 'no', 'ili' 라는 쓸모없는 단어들이 맨앞에 나오는 경우, 그 외 단어들도 arr1에 저장한다. upper 함수를 통해 arr1에 저장된 단어들을 대문자로 출력한다.