[백준 알고리즘][파이썬/Python] 1152번: 단어의 개수

문제

영어 대소문자와 띄어쓰기만으로 이루어진 문자열이 주어진다. 이 문자열에는 몇 개의 단어가 있을까? 이를 구하는 프로그램을 작성하시오. 단, 한 단어가 여러 번 등장하면 등장한 횟수만큼 모두 세어야 한다.

 

입력

첫 줄에 영어 대소문자와 띄어쓰기로 이루어진 문자열이 주어진다. 이 문자열의 길이는 1,000,000을 넘지 않는다. 단어는 띄어쓰기 한 개로 구분되며, 공백이 연속해서 나오는 경우는 없다. 또한 문자열의 앞과 뒤에는 공백이 있을 수도 있다.

 

출력

첫째 줄에 단어의 개수를 출력한다.

 

입출력 예

 

제출 코드

s = input().strip()
if len(s) == 0:
    print(0)
else:
    print(s.count(' ')+1)
    
# 실패 코드
# lst = input().strip().split(' ')
# print(len(lst))

코드가 너무 짧아서 라인 단위로 끊어 설명은 딱히 안 하겠다. 어이없이 실패가 몇 번 났었는데, 입력값이 단어 한 개도 없이 공백으로 주어지는 경우가 있다는 것을 고려하지 못했다. 그래서 원래는 7번, 8번 라인같이 짰었는데, 2번 라인처럼 예외를 걸어줘야 한다는 것이다.

 

 

 

Source : https://www.acmicpc.net/problem/1152