[백준/BOJ] 8958번_OX퀴즈 (C++/Java)

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퀴즈의 결과가 주어졌을 때, 점수를 구하는 프로그램을 작성하시오.

입력

---

첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스는 한 줄로 이루어져 있고, 길이가 0보다 크고 80보다 작은 문자열이 주어진다. 문자열은 O와 X만으로 이루어져 있다.

출력

---

각 테스트 케이스마다 점수를 출력한다.

💻 예제 입력

5
OOXXOXXOOO
OOXXOOXXOO
OXOXOXOXOXOXOX
OOOOOOOOOO
OOOOXOOOOXOOOOX

💻 예제 출력

10
9
7
55
30

풀이

---

문자열을 string으로 받는다.

O가 연속해서 나오면 1씩 누적된 값을 더해주어야 하므로 누적합을 담을 ac 변수를 두고, 반복문으로 문자열을 하나씩 검사하면서 O가 나올 경우 ac에 1을 더한 후 sum에 더한다.

X가 나올 경우 ac를 0으로 갱신한다.

만약 입력된 문자열이 OOXO일 경우, 첫번째 O에서 ac = 1, sum = 1이 되고, 두번째 O에서는 ac = 2, sum = 3이 된다.

세번째에 X가 나오는 순간 ac = 0이 되고, 그 다음 O가 들어오면 ac = 1, sum = 4가 된다.

C++ 소스코드

---

#include <iostream>
using namespace std;

int main(void) 
{
    int N;
    cin >> N;
    
    while (N--) 
		{
        int sum = 0, ac = 0;
        string str;
        cin >> str;

        for (int i = 0; i < str.length(); i++) 
				{
            if (str[i] == 'O') {
                ac += 1;  // 누적합
                sum += ac;
            }
            else ac = 0;
        }
        cout << sum << "\n";
    }
}

Java 소스코드

---

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));
        int N = Integer.parseInt(br.readLine());

        while (N-- != 0)
        {
            String str = br.readLine();
            int sum = 0, ac = 0;

            for (int i = 0; i < str.length(); i++)
            {
                if (str.charAt(i) == 'O') {
                    ac += 1;
                    sum += ac;
                } else
                    ac = 0;
            }
            System.out.println(sum);
        }
    }
}