BAEKJOON 문제 풀이 : 8958번 OX퀴즈

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

8958번: OX퀴즈

 

인풋을 배열로 입력받아 문자 'O'가 연속된 횟수를 더하는 문제입니다.

파이썬에서 list와 for는 떼려야 뗄 수 없는 관계라는걸 다시 한 번 생각하게된 문제였습니다.

문제 풀이는 아래와 같습니다.

 

1. Java

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
 
public class Main {
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        
        int cntLoop = Integer.parseInt(br.readLine());
        for (int i = 0; i < cntLoop; i++) {
            char[] arrChar = br.readLine().toCharArray();
            int score = 0;
            int cntContinuity = 0;
            
            for (int j = 0; j < arrChar.length; j++) {
                if (arrChar[j] == 'O') {
                    cntContinuity++;
                    score += cntContinuity;
                    
                } else {
                    cntContinuity = 0;
                }
            }
            
            bw.write(score + "\n");
        }
        
        bw.close();
        br.close();
    }
}

 

2. Python 3

import sys
 
if __name__ == '__main__':
    cntLoop = int(sys.stdin.readline())
    for i in range(cntLoop):
        inputData = sys.stdin.readline()
        score = [0] * (len(inputData) + 1)
         
        for j in range(len(inputData)):
            if (inputData[j] == 'O'):
                score[j + 1] = score[j] + 1
          
        print(sum(score))

 

아래는 조금 더 짧게 줄인 코드입니다.

import sys
 
if __name__ == '__main__':
    for i in range(int(sys.stdin.readline())):
        print(sum(sum(range(1, len(j) + 1)) for j in sys.stdin.readline().strip().split('X')))