백준 1152번 단어의 개수 - SWIFT, C++

문제

---

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

 

입력

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

 

출력

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

 

문제 링크

1152번: 단어의 개수

1152번: 단어의 개수

 

 

풀이

---

Swift로 풀면 2줄만에 풀리는 자존심 상하는 문제다.

코딩을 막 시작했을 때, C++로 풀었는데 그때는 앞과 뒤의 공백을 생각하지 않아서 틀렸었다.

문제 조건이 단어는 띄어쓰기 한 개로 구분되며, 공백이 연속해서 나오는 경우는 없다. 이기 때문에 일반적인 상황이라면 공백의 수 + 1이 단어의 수가 된다.

하지만 앞 또는 뒤에 공백이 올 수도 있기 때문에 앞 또는 뒤의 공백은 제거하고 세주어야 한다.▼

 

 

Swift 코드

---

var arr = readLine()!.split(separator: " ")
print(arr.count)

 

잘 우겨넣으면 한 줄로도 가능하긴 할거다.

separator로 들어간 문자열을 기준으로 문자열을 잘라서 arr에 넣어준다.

지금은 공백이 들어갔으므로, 공백을 기준으로 앞 뒤 문자열을 잘라서 arr에 넣게 된다.

그러면 단어의 수만 들어가게 될 것이고, 배열의 요소 개수를 세주면 답이 된다.

separator의 경우에는 이 문제에서만 특별하게 잠깐 나온게 아니고, 다른 문제들에서 공백을 통해 입력이 들어올 때도 유용하게 사용할 수 있다.

 

 

C++ 코드

---

#include<iostream>
#include<string>
using namespace std;

int main()
{
	string line;
	getline(cin, line, '\n');
	int n = 1;
	for (int i = 0; i < line.length(); i++)
	{
		if (i != 0 )
		{
			if(line.at(i)==' ' && i !=line.length()-1)
			n++;
		}
		else if (line.at(0) == ' ')
		{
			if (line.length() == 1)
				n = 0;
		}
		
	}
	cout << n;
	return 0;
}