백준 10820 문자열 분석 (Java)

URL

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

10820번: 문자열 분석

문제

문자열 N개가 주어진다. 이때, 문자열에 포함되어 있는 소문자, 대문자, 숫자, 공백의 개수를 구하는 프로그램을 작성하시오.

각 문자열은 알파벳 소문자, 대문자, 숫자, 공백으로만 이루어져 있다.

입력

첫째 줄부터 N번째 줄까지 문자열이 주어진다. (1 ≤ N ≤ 100) 문자열의 길이는 100을 넘지 않는다.

출력

첫째 줄부터 N번째 줄까지 각각의 문자열에 대해서 소문자, 대문자, 숫자, 공백의 개수를 공백으로 구분해 출력한다.

성공 코드

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));
        StringBuilder sb;

        String input;
        int[] count;

        while ((input = br.readLine()) != null) {
            count = new int[4];

            sb = new StringBuilder();

            for (int i=0; i<input.length(); i++) {
                int code = input.charAt(i);

                if (code >= 97 && code <= 122) {
                    count[0] += 1;
                }
                if(code >= 65 && code <= 90) {
                    count[1] += 1;
                }
                if(code >= 48 && code <= 57) {
                    count[2] += 1;
                }
                if(code == 32) {
                    count[3] += 1;
                }
            }

            for(int i : count) {
                sb.append(i).append(" ");
            }

            System.out.println(sb);
        }
    }
}