인프런 커뮤니티 질문&답변

안녕하세요 선생님

덕분에 코딩테스트 열심히 준비하고 있는 학생입니다.

다름이 아니라 궁금한 점이 있어 이렇게 문의 드립니다.

강의와 같은 코드를 구현 후 로컬에서 실행 시켰을 경우에는 답이 정확하게 나오는데 채점 사이트에 올리면 오답이라고 답이 계속 나오는데 혹시 제가 잘못한 부분인건지, 아니면 채점 사이트에서 요구하는 것이 잘못된 것인지 여쭤보고 싶습니다.

import java.util.*;
class Main {
    public int solution(int number, int[] numArray) {
        int answer = 0;
        int cnt = 0;
        for (int i = 0; i < number - 1; i++) {
            if (numArray[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);
        Integer number = kb.nextInt();
        int[] numArray = new int[number];
        for (int i = 0; i < number; i++) {
            numArray[i] = kb.nextInt();
        }
        System.out.print(T.solution(number, numArray));
    }
}