[BOJ] 8958번 OX퀴즈

백준#8958 OX퀴즈
https://www.acmicpc.net/problem/8958

코드 - Python

for _ in range(int(input())):
    score = 0
    cnt = 1
    for a in input():
        if a == "O":
            score += cnt
            cnt +=1
        else:
            cnt = 1
    print(score)

코드 - Java

import java.util.*;
public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        for (int i=0;i<n;i++){
            int score = 0;
            int cnt = 1;
            String[] str = sc.next().split("");
            for (String s:str){
                if (s.equals("O")){
                    score+=cnt;
                    cnt++;
                }
                else cnt = 1;
            }

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