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

강의보기전 짜본 코드입니다. 이클립스에서 실행할때는 답이 맞게 나오는데 채점 사이트에서는 에러가 출력되어 질문드립니다.

package codingTest;

import java.util.*;

public class Main {

	public int Solution(int n, int[] ns) {
		int answer = 0;
		int reward = 0;

		for (int i = 0; i < n; i++) {
			if (ns[i] == 1) {
				reward++;
				answer += reward;
			} else if (ns[i] == 0) {
				reward = 0;
				answer += reward;
			}
		}

		return answer;
	}

	public static void main(String[] args) {

		// TODO Auto-generated method stub
		Main T = new Main();
		Scanner kb = new Scanner(System.in);

		int n = kb.nextInt();
		int[] ns = new int[n];

		for (int i = 0; i < n; i++) {
			ns[i] = kb.nextInt();
		}

		System.out.println(T.Solution(n, ns));

	}

}