[goormlevel/Java] 퀴즈 대회

-Code

import java.io.*;
import java.util.*;

class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		// 문제 갯수
		int cnt = Integer.parseInt(br.readLine());
		// 각 문제가 맞았는지 확인
		StringTokenizer st = new StringTokenizer(br.readLine());
		// 총 결과와 현재 추가되는 점수 초기화
		int totalScore = 0, currScore = 1;
		for (int i = 0; i < cnt; i++) {
			int isCorrect = Integer.parseInt(st.nextToken());
			// 틀렸다면 currScore을 1로 초기화
			if (isCorrect == 0) {
				currScore = 1;
				continue;
			}
			// 맞았다면 추가되는 currScore 더하고 +1
			totalScore += currScore;
			currScore++;
		}
		br.close();
		System.out.println(totalScore);
	}
}