[백준] 1152번: 단어의 개수(java, python)

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

1152번: 단어의 개수

---

💡 주의 사항

공백 입력 시, 1이 아닌 0 이 출력되야 함 - 공백 입력 후 split()으로 처리했을 때 결과는?

---

🗒 풀이

 java

 

사실 첫 번째 시도는 실패했다.

주의사항을 제대로 인지하지 못하고, split() 함수를 사용해 제출했었는데, 공백 입력 시 1이 출력된다는 것을 간과했던 것이다.

 

틀린 코드는 다음과 같다.

 

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String str = sc.nextLine();
        String[] strList = str.split(" ");
        System.out.println(strList.length);
    }
}

1) split() 이용

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String str = sc.nextLine();
        str = str.trim();
        if (str.isEmpty()) {
            System.out.println(0);
        } else {
            String[] strList = str.split(" ");
            System.out.println(strList.length);
        }
    }
}

• 공백 입력 시 1이 출력됨을 방지하기 위해 trim() 함수로 공백 처리
• 공백 처리 후, str 변수가 비어있다면 공백만 입력된 것이므로 0을 출력하고, 그 외에는 split()을 이용하여 처리

 

2) StringTokenizer 이용

import java.util.Scanner;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String str = sc.nextLine();
        StringTokenizer st = new StringTokenizer(str, " ");
        System.out.println(st.countTokens());
        sc.close();
    }
}

• split()이 아닌 StringTokenizer을 이용하여 처리
• StringTokenizer 사용 시, 분리 기준을 공백으로 지정해주면 공백을 기준으로 나뉘어 토큰에 저장되기 때문에 공백 예외처리를 따로 해주지 않아도 됨
• StringTokenizer의 countTokens()를 사용하여 출력

---

python

s = input()
print(len(s.strip().split()))

• 입력받은 문자열 s를 strip() 함수를 이용해 공백 처리 후, 공백 기준으로 split() 한 것의 길이를 출력

---