[백준 8958] OX 퀴즈 (C++/Python)

C++)

#include <iostream>
#include <string>
using namespace std;

int main(void) {

	ios_base::sync_with_stdio(false);
	cin.tie(nullptr);

	int t;
	cin >> t;

	while (t--) {
		string str;
		cin >> str;

		int answer = 0;
		int temp = 0;
		
		for (auto ch : str) {
			if (ch == 'O') {
				temp += 1;
			}
			else {
				temp = 0;
			}
			answer += temp;
		}

		cout << answer << "\n";
	}

	return 0;
}

 

 

 

Python)

import sys

read = sys.stdin.readline

t = int(read().rstrip())

for _ in range(t):
    result = read().rstrip()
    
    answer = 0
    temp = 0
    
    for ch in result:
        if ch == 'O':
            temp += 1
        else :
            temp = 0
        answer += temp
    
    print(answer)