백준 - 8958번(OX퀴즈)

문제 출처: https://www.acmicpc.net/problem/8958

문제

• "OOXXOXXOOO"와 같은 OX퀴즈의 결과가 있다. O는 문제를 맞은 것이고, X는 문제를 틀린 것이다. 문제를 맞은 경우 그 문제의 점수는 그 문제까지 연속된 O의 개수가 된다. 예를 들어, 10번 문제의 점수는 3이 된다.

• "OOXXOXXOOO"의 점수는 1+2+0+0+1+0+0+1+2+3 = 10점이다.

• OX퀴즈의 결과가 주어졌을 때, 점수를 구하는 프로그램을 작성하시오.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(reader.readLine());
        StringBuilder sb = new StringBuilder();

        while (N-- > 0) {
            char[] target = reader.readLine().toCharArray();
            int sum = 0;
            int score = 0;
            for (char c : target) {
                if (c == 'O')
                    sum += ++score;
                else
                    score = 0;
            }
            sb.append(sum).append("\n");
        }

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

}

• BufferedReader를 통해 입력스트림을 받을 준비를 하고 먼저 첫째줄에 입력되는 라인 수를 저장한다.
• 라인 수만큼 반복을 수행하며 입력되는 값을 받는다.
• 누적합이 존재하므로 전체 합을 저장하는 sum 변수와 현재 점수를 저장하는 score를 둔다.
• 입력된 값의 길이만큼 반복하며 값이 'O'일 경우 값을 누적함과 동시에 전체 합계에 더한다.
• 'X'를 만났을 경우, 현재 점수를 0으로 수정한다.