백준 2506 java

벡터 연습하게 벡터에 저장하고 다시 읽어서 정답을 찾는 방식으로 할 걸 그랬다...

import java.util.Scanner;

public class bj2506 {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int N;
        int i, j = 0, sum = 0, correct;

        N = scanner.nextInt();
        for(i = 0; i < N; i++)
        {
            correct = scanner.nextInt();
            if(correct == 1)
            {
                j++;
                sum += j;
            }
            else
            {
                j = 0;
            }
        }
        System.out.println(sum);

        scanner.close();
    }
}