※개인 공부 목적의 정리글입니다.
이 글의 내용이 최선의 해답은 아닐 수 있습니다.
문제
영어 대소문자와 공백으로 이루어진 문자열이 주어진다. 이 문자열에는 몇 개의 단어가 있을까? 이를 구하는 프로그램을 작성하시오. 단, 한 단어가 여러 번 등장하면 등장한 횟수만큼 모두 세어야 한다.
입력
첫 줄에 영어 대소문자와 공백으로 이루어진 문자열이 주어진다. 이 문자열의 길이는 1,000,000을 넘지 않는다. 단어는 공백 한 개로 구분되며, 공백이 연속해서 나오는 경우는 없다. 또한 문자열은 공백으로 시작하거나 끝날 수 있다.
출력
첫째 줄에 단어의 개수를 출력한다.
요약
문제 자체는 쉬워보인다. 단순히 앞뒤 공백을 날리고(trim) 공백 기준으로 split을 한 후 나온 배열의 길이를 출력하면 될거라고 생각했었는데…
주의할 점이 있다. 자세한 것은 아래서 설명하겠다.
나의 코드(실패)
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
public class Main {
public static void main(String[] args) throws IOException{
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
String str = br.readLine().trim();
System.out.println(str.split(" ").length);
br.close();
}
}단순히 입력을 trim하여 앞뒤 공백을 날린 후 split 한 결과의 길이를 출력했더니 결과는 실패!
무엇이 문제였는지 생각해보았으나 떠오르지가 않아서 결국 검색 찬스를 썼다…
그 결과 바로 입력이 공백만 있는 경우!
당연히 문장이 들어올거라고 생각했는데 예상치못한 케이스였다.
확실히 공백만 들어오면 trim되면서 문자열은 ” “에서 빈 문자열인 “”이 되어버리고, 이 상태에서 공백을 기준으로 split을 하게되면 빈문자열 그 자체만 하나의 원소로 가지는 배열이 반환되므로 그 배열의 길이인 1이 출력되어 틀리는 것이었다.
나의 코드(성공)
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
public class Main {
public static void main(String[] args) throws IOException{
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
String str = br.readLine().trim();
if(str.equals("")){
System.out.println(0);
}else{
System.out.println(str.split(" ").length);
}
br.close();
}
}해결방법은 간단하다.
입력받은 문자열을 trim한 후 split 하기 전에 문자열이 비어있으면 0을 출력하도록 하면 된다
댓글