19. 점수계산

import java.util.*;

class Main {    
    public int solution(int n, int[] arr) {
        int answer = 0;
        int cnt = 0;
        for(int i = 0; i < n; i++) {
            if(arr[i] == 1) {
            	// i가 1이면 cnt+1하고 answer에 누적
                cnt++;
                answer += cnt;
            } else {
            	// i가 1이 아니면 cnt 초기화
                cnt = 0;
            }
        }
        return answer;
    }

    public static void main(String[] args){
        Main T = new Main();
        Scanner kb = new Scanner(System.in);
        int n = kb.nextInt();
        int[] arr = new int[n];
        for(int i = 0; i < n; i++) {
            arr[i] = kb.nextInt();
        }
        System.out.println(T.solution(n, arr));

    }
}

import java.util.*;

class Main {    

    public int solution(int n, int[] arr) {

        int answer = 0;

        int cnt = 0;

        for(int i = 0; i < n; i++) {

            if(arr[i] == 1) {

                cnt++;

                answer += cnt;

            } else {

                cnt = 0;

            }

        }

        return answer;

    }

    public static void main(String[] args){

        Main T = new Main();

        Scanner kb = new Scanner(System.in);

        int n = kb.nextInt();

        int[] arr = new int[n];

        for(int i = 0; i < n; i++) {

            arr[i] = kb.nextInt();

        }

        System.out.println(T.solution(n, arr));

    }

}