점수 계산하기

package main;



import java.util.ArrayList;
import java.util.Scanner;


public class Main {


	public  int  sol( int n , ArrayList<Integer> array) {
	
		ArrayList<Integer> answer = new ArrayList<>();
		int count = 0;
		
		//반복문 돌려서
		for(int i  = 0; i < n; i ++) {
			// 0 이면 count를 0으로 초기화시켜서 answer 배열에 넣어준다 
			if(array.get(i) == 0) {
				count = 0;
				answer.add(count);
				
			// 0 이 아니면 count 1씩 추가 시켜서 값을 answer 배열에 넣어준다	
			}else if(array.get(i) != 0) 
				count ++;
				answer.add(count);
			
			}	
		
		int sum = 0;
		
		// 값을 담은 배열안에 있는 값 전체를 더해준다
		for(int i = 0 ; i < answer.size(); i ++) {
		  sum += answer.get(i);
		}
		
		//값을 도출한다
		return sum ;
		}

	
	
	
	public static void main(String[] args) {
		
	
	
	Main main = new Main();
	Scanner scan = new Scanner(System.in);
	
	int b  = scan.nextInt();
	
	ArrayList<Integer> array = new ArrayList<>();
	
	for(int i = 0; i < b; i ++) {
		array.add(scan.nextInt());
	}
	
	System.out.println(main.sol(b,array));

	}
}