[Java] 백준 8958번

Input-1)
5
OOXXOXXOOO
OOXXOOXXOO
OXOXOXOXOXOXOX
OOOOOOOOOO
OOOOXOOOOXOOOOX

Output-1)
10
9
7
55
30

접근 방식

OX 퀴즈 결과 문자열을 순회하며 아래의 조건을 통해 점수를 계산하였다.

• 1) O인 경우 점수를 1 증가시킨다.
• 2) X인 경우 점수를 증가시키지 않는다.
• 3) 점수가 계산될 떄마다 해당 점수를 합한다.

하나의 OX퀴즈 문자열 점수 계산이 끝나고 합을 구했다면 4) 별도의 합 점수 배열에 담는다.

for (int i=0; i<n; i++) {
    ox_arr = br.readLine().split(""); // 입력받을 OX퀴즈 문자열
    sum = 0; // 문자열 점수들의 합
    score = 0; // 해당 문자열의 점수
    for (int j=0; j<ox_arr.length; j++) {
        if(ox_arr[j].equals("O")) { // 1번
            score += 1; 
            sum += score; // 3번
        }
        else { // 2번
            score = 0;
            sum += score; // 3번
        }
    }
    ox_scores[i] = sum; // 4번
}

작성코드

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

class Main {
    public static void main(String[] args) throws IOException {
        
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int n = Integer.parseInt(br.readLine());
        int sum, score;
        String[] ox_arr = new String[n];
        int[] ox_scores = new int[n];

        for (int i=0; i<n; i++) {
            ox_arr = br.readLine().split("");
            sum = 0;
            score = 0;
            for (int j=0; j<ox_arr.length; j++) {
                if(ox_arr[j].equals("O")) {
                    score += 1;
                    sum += score;
                }
                else {
                    score = 0;
                    sum += score;
                }
            }
            ox_scores[i] = sum;
        }

        for (int ox_score : ox_scores) bw.write(ox_score + "\n");
        
        bw.flush();
        bw.close();
        br.close();
    }
}